Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
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 <[email protected]>
- Loading branch information