Add functionality to restore model training from the latest model check-point.
Add functionality to restore model training from the latest model check-point.