-
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
79c1bad7
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