gluon-cv icon indicating copy to clipboard operation
gluon-cv copied to clipboard

Resuming from checkpoints for classification scripts

Open d-li14 opened this issue 7 years ago • 3 comments

Hi, thanks for the great code!

I wonder do you have plans to support resuming from checkpoints for classification? As we all know, in terms of training ImageNet, the training process is really long and it can be interrupted somehow, but I haven't notice any code related to "resume" in scripts/classification/train_imagenet.py.

Maybe @hetong007 ? Thanks in advance.

d-li14 avatar Sep 24 '18 10:09 d-li14

Will add that feature soon.

In case you need it right now, you can save both the parameters by net.save_parameters(), and the state of the trainer by trainer.save_states(), and load them in when resuming.

hetong007 avatar Sep 24 '18 18:09 hetong007

OK, thanks for adding this new feature.

d-li14 avatar Sep 25 '18 13:09 d-li14

Como me llega el código de liberación

jorgetony253 avatar Apr 26 '20 06:04 jorgetony253