Feature to resume training after stopping?

[pracdl314]

Hi @ZhihaoDU, does this repository have a feature to resume training after stopping after a certain number of epochs or in the middle of a epoch? Thanks in advance!

[ZhihaoDU]

sure! please check if there is an option like --resume true. If there is such option in the run.sh, the trainer will resume the training process automatically by loading model parameters and optimizer states. If there is such option, just add it.