Commit Graph
2 Commits
Author SHA1 Message Date
bilzardandGlenn Jocher aff0281969 Load checkpoint on CPU instead of on GPU (#6516)
* Load checkpoint on CPU instead of on GPU

* refactor: simplify code

* Cleanup

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-04 18:58:18 +01:00
bilzard e1dc894364 Enable AdamW optimizer (#6152) 2022-01-02 13:10:19 -08:00