ultralytics-ascend/ultralytics/yolo/engine
Neloy262 586c95b8a9
Add torch.Tensor results image saving (#3475)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-03 16:08:02 +02:00
..
__init__.py Add yolo package structure (#15) 2022-09-19 15:47:23 +05:30
exporter.py Fix Pose export from YAML (#3312) 2023-06-21 18:04:22 +02:00
model.py Docstrings arguments cleanup (#3229) 2023-06-17 16:17:07 +02:00
predictor.py Add torch.Tensor results image saving (#3475) 2023-07-03 16:08:02 +02:00
results.py Add torch.Tensor results image saving (#3475) 2023-07-03 16:08:02 +02:00
trainer.py Increase NCCL timeout from 1 hour to 3 hours (#3343) 2023-06-24 15:56:18 +02:00
validator.py Move loss calculation to head (#2874) 2023-05-30 19:24:30 +02:00