yolov5/utils
Glenn Jocher ca19df5f7f
Add `autobatch` feature for best `batch-size` estimation (#5092)
* Autobatch

* fix mem

* fix mem2

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update train.py

* print result

* Cleanup print result

* swap fix in call

* to 64

* use total

* fix

* fix

* fix

* fix

* fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Cleanup printing

* Update final printout

* Update autobatch.py

* Update autobatch.py

* Update autobatch.py
2021-10-25 13:56:13 +02:00
..
aws Fix `yaml.safe_load()` ignore emoji errors (#5060) 2021-10-05 13:41:52 -07:00
flask_rest_api Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
loggers W&B: Media panel fix (#5317) 2021-10-25 12:06:12 +02:00
__init__.py Update `sys.path.append(str(ROOT))` (#4852) 2021-09-18 15:02:08 +02:00
activations.py Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
augmentations.py Update default Albumentations (#4931) 2021-09-26 12:10:33 -07:00
autoanchor.py Refactor `Detect()` anchors for ONNX <> OpenCV DNN compatibility (#4833) 2021-10-11 09:58:42 -07:00
autobatch.py Add `autobatch` feature for best `batch-size` estimation (#5092) 2021-10-25 13:56:13 +02:00
callbacks.py Optimised Callback Class to Reduce Code and Fix Errors (#4688) 2021-09-07 18:32:15 +02:00
datasets.py take EXIF orientation tags into account when fixing corrupt images (#5270) 2021-10-20 21:35:47 +02:00
downloads.py Update autodownload fallbacks to v6.0 assets (#5177) 2021-10-13 19:58:38 -07:00
general.py Improved check_suffix() robustness to `''` and `""` (#5192) 2021-10-14 12:00:39 -07:00
loss.py Remove redundant `ComputeLoss` code (#4701) 2021-09-08 12:48:33 +02:00
metrics.py Adjust legend labels for classes without instances (#5174) 2021-10-14 11:57:00 -07:00
plots.py Update plots.py (#5127) 2021-10-11 01:59:22 -07:00
torch_utils.py Add `autobatch` feature for best `batch-size` estimation (#5092) 2021-10-25 13:56:13 +02:00