Resuming from checkpoints for classification scripts
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.
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.
OK, thanks for adding this new feature.
Como me llega el código de liberación