Updating logic for `--restart_from_checkpoint`
The purpose of this MR is to update the logic for --restart_from_checkpoint
such that the --gw
flag is not needed.
The purpose of this MR is to update the logic for --restart_from_checkpoint
such that the --gw
flag is not needed.