You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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.
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.
The text was updated successfully, but these errors were encountered: