EMA changes for pre-model's batch_size (#3681)
* EMA changes for pre-model's batch_size * Update train.py * Update torch_utils.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>pull/3684/head
parent
f527704cd3
commit
2729761458