Add restore_steps to training script Args #224
Description
Currently, checkpointmanager.latest_step is hardcoded into our training scripts. We should expose this to the Args.
{{ message }}
Add
restore_steps to training script Args #224Currently, checkpointmanager.latest_step is hardcoded into our training scripts. We should expose this to the Args.