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.
Showing
Please
register
or
sign in
to comment