Fix tot update in trainer (#37923)

Commit

222 days ago

Fix tot update in trainer (#37923) * fix total updates in epoch * add test; fix max_steps * replace with multi-gpu decorator