Commit Graph
5 Commits
Author SHA1 Message Date
Colin Wong f4b05680f8 Assert --optimize not used with cuda device (#8569) 2022-07-15 16:01:01 +02:00
Colin Wong f8722b4429 Raise error on suffix-less model path (#8561)
Raise error on invalid model
2022-07-13 11:13:01 +02:00
526e650553 Fix LoadImages() with dataset YAML lists (#8517)
* Fix LoadImages with dataset yaml lists

* Update dataloaders.py

* Update dataloaders.py

* Simplify/refactor PR

* Update dataloaders.py

Co-authored-by: Colin Wong <noreply@brains4drones.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-08 14:32:40 +02:00
Colin WongandGlenn Jocher be42a24d23 Properly expose batch_size from OpenVINO similarly to TensorRT (#8514)
Properly expose `batch_size` from OpenVINO

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-07 23:53:09 +02:00
Colin Wong da2ee3934e Expose OpenVINO batch_size similarly to TensorRT (#8437) 2022-07-01 22:15:09 +02:00