yolov5/utils
NanoCode012 035ac82ed0
Fix torch multi-GPU --device error (#1701)
* Fix torch GPU error

* Update torch_utils.py

single-line device =

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-12-15 20:42:14 -08: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 FReLU bias=False bug fix (#1666) 2020-12-10 13:06:15 -08:00
autoanchor.py Update matplotlib.use('Agg') tight (#1583) 2020-12-02 15:53:16 +01:00
datasets.py Streaming --save-txt bug fix (#1672) 2020-12-11 15:45:32 -08:00
general.py Simplify autoshape() post-process (#1653) 2020-12-09 07:44:06 -08:00
google_utils.py vast.ai compatability updates (#1657) 2020-12-09 19:01:08 -08:00
loss.py Add QFocalLoss() (#1482) 2020-11-25 19:32:27 +01:00
metrics.py Reinstate PR curve sentinel values (#1645) 2020-12-08 17:40:49 -08:00
plots.py Add idetection_profile() function to plots.py (#1700) 2020-12-15 18:35:47 -08:00
torch_utils.py Fix torch multi-GPU --device error (#1701) 2020-12-15 20:42:14 -08:00