yolov5_ZWgra

1,530 Commits 18 Branches 0 Tags

2 Commits

Author	SHA1	Message	Date
Jebastin NadarandGlenn Jocher	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
Jebastin Nadar	1492632796	Fix zipfile name for coco128-segments (#4914 )	2021-09-25 06:43:24 -07:00

Author

SHA1

Message

Date

Jebastin NadarandGlenn Jocher

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

Jebastin Nadar

1492632796

Fix zipfile name for coco128-segments (#4914 )

2021-09-25 06:43:24 -07:00