Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
train.py GPU memory fix (ultralytics#3590)
* train.py GPU memory fix * ema * cuda * cuda * zeros input * to device * batch index 0 (cherry picked from commit 4984cf5)
- Loading branch information