You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Neloy262 586c95b8a9
Add torch.Tensor results image saving (#3475)
2 years ago
..
__init__.py Add yolo package structure (#15) 2 years ago
exporter.py Fix Pose export from YAML (#3312) 2 years ago
model.py Docstrings arguments cleanup (#3229) 2 years ago
predictor.py Add torch.Tensor results image saving (#3475) 2 years ago
results.py Add torch.Tensor results image saving (#3475) 2 years ago
trainer.py Increase NCCL timeout from 1 hour to 3 hours (#3343) 2 years ago
validator.py Move loss calculation to head (#2874) 2 years ago