Add ability to restore the state from the latest checkpoint #78
Description
Add functionality to restore model training from the latest model check-point.
{{ message }}
Add functionality to restore model training from the latest model check-point.