yolov5/utils
Glenn Jocher d8f18834a2
Update `profile()` for CUDA Memory allocation (#4239)
* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Cleanup
2021-07-30 18:17:19 +02:00
..
aws Update resume.py (#4115) 2021-07-22 17:22:11 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +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 PyCharm reformat (#4209) 2021-07-28 23:35:14 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
augmentations.py Albumentations >= 1.0.3 (#4068) 2021-07-19 13:23:19 +02:00
autoanchor.py Update script headers (#4163) 2021-07-26 15:23:33 +02:00
datasets.py Train from `--data path/to/dataset.zip` feature (#4185) 2021-07-28 02:04:10 +02:00
downloads.py PyCharm reformat (#4209) 2021-07-28 23:35:14 +02:00
general.py Add `@try_except` decorator (#4224) 2021-07-29 17:23:35 +02:00
loss.py New CSV Logger (#4148) 2021-07-25 19:06:37 +02:00
metrics.py Copy-Paste augmentation for YOLOv5 (#3845) 2021-07-01 00:35:04 +02:00
plots.py New CSV Logger (#4148) 2021-07-25 19:06:37 +02:00
torch_utils.py Update `profile()` for CUDA Memory allocation (#4239) 2021-07-30 18:17:19 +02:00