BIT-DA/SePiCo

Could you point out where is the code for saving model parameters in the sepico repository? So that I can resume training from breakpoints?

ZHE-SAPI opened this issue · 1 comments

ZHE-SAPI commented

Where is the code for saving model parameters, so that it is convenient to resume training from breakpoints?

ZHE-SAPI commented

Oh, I got it. Just like this.

cfg.checkpoint_config = dict(
by_epoch=False,
interval=1000,
max_keep_ckpts=5,
out_dir='path/to/checkpoints'
)