-
Ruoxin Sang authored
Don't early return if the global step is not updated by `num_steps` as expected. This still allows summary and metric reporting. PiperOrigin-RevId: 386967587
b26a5ca6
Don't early return if the global step is not updated by `num_steps` as expected. This still allows summary and metric reporting. PiperOrigin-RevId: 386967587