yolov5/utils
yujun 05a955a3f6
FLOPS computation device bug fix (#1447)
* Update torch_utils.py

fix issue#113 , inputs device should be same with model parameters' device

* Update torch_utils.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-19 12:56:20 +01:00
..
google_app_engine adding the configuration to deploy yolov5 in in app engine (#964) 2020-09-22 11:48:44 -07:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py Utils reorganization (#1392) 2020-11-14 11:50:32 +01:00
autoanchor.py Utils reorganization (#1392) 2020-11-14 11:50:32 +01:00
datasets.py Update img2label_paths() (#1414) 2020-11-17 13:26:06 +01:00
general.py PyTorch Hub and autoShape update (#1415) 2020-11-16 23:09:55 +01:00
google_utils.py Utils reorganization (#1392) 2020-11-14 11:50:32 +01:00
loss.py Utils reorganization (#1392) 2020-11-14 11:50:32 +01:00
metrics.py Update PR curve (#1428) 2020-11-17 17:29:55 +01:00
plots.py Update labels.png with rectangles fix (#1432) 2020-11-18 12:27:30 +01:00
torch_utils.py FLOPS computation device bug fix (#1447) 2020-11-19 12:56:20 +01:00