ultralytics-ascend/ultralytics/engine
Jiacong Fang 03d0ffd9f5
Improved float workspace arg for TRT exports (#9407)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-29 16:22:32 +01:00
..
__init__.py Implement all missing docstrings (#5298) 2023-10-10 20:07:13 +02:00
exporter.py Improved float workspace arg for TRT exports (#9407) 2024-03-29 16:22:32 +01:00
model.py ultralytics 8.1.37 fix empty sys.argv bug (#9390) 2024-03-29 03:20:39 +01:00
predictor.py Faster batch same_shapes (#8851) 2024-03-11 19:53:35 +01:00
results.py ultralytics 8.1.34 Inference API robust imgsz checks (#9274) 2024-03-25 02:00:38 +01:00
trainer.py ultralytics 8.1.36 improve train stop robustness epoch + 1 >= self.epochs (#9384) 2024-03-28 17:28:20 +01:00
tuner.py Add bgr hyperparameter (#9139) 2024-03-23 23:17:00 +01:00
validator.py Integrate OpenVINO CUMULATIVE_THROUGHPUT mode batched inference (#8834) 2024-03-11 18:50:29 +01:00