EMA changes for pre-model's batch_size (#3681)
Browse files* EMA changes for pre-model's batch_size
* Update train.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <[email protected]>
* EMA changes for pre-model's batch_size
* Update train.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <[email protected]>