Load checkpoint on CPU instead of on GPU#6516

Merged

glenn-jocher merged 4 commits into

ultralytics:masterfrom

bilzard:patch-2

Feb 4, 2022

Commits on Feb 4, 2022

Load checkpoint on CPU instead of on GPU

bilzard
authored and
glenn-jocher
committed
refactor: simplify code

bilzard
authored and
glenn-jocher
committed
Cleanup
glenn-jocher
authored
Update train.py
glenn-jocher
authored