NanoCode012andGlenn Jocher 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