zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
yolov5/models
History
Jebastin Nadar a0e15046c3
Fix different devices bug when moving model from GPU to CPU (#5110) 
* fix different devices bug

* extend _apply() instead of to() for a general fix

* Only apply if Detect() is last layer

Co-authored-by: Jebastin Nadar <njebastin10@gmail.com>

* Indent fix

* Add comment to yolo.py

* Add comment to common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 		2021-10-10 14:24:05 -07:00
..
hub Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs (#4608) 2021-08-30 15:46:33 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Fix different devices bug when moving model from GPU to CPU (#5110) 2021-10-10 14:24:05 -07:00
experimental.py Add TensorFlow and TFLite export (#1127) 2021-08-17 13:18:16 +02:00
tf.py Fix pylint: do not use bare 'except' (#5025) 2021-10-03 17:54:40 -07:00
yolo.py Fix different devices bug when moving model from GPU to CPU (#5110) 2021-10-10 14:24:05 -07:00
yolov5l.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5m.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5s.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5x.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00