Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add auto save ckpt at the end of training (#260)
Currently, checkpoints will not be saved until the training steps satisfy the saving frequency. This PR adds an auto-save ckpt function at the end of training.
- Loading branch information