Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Resuming from checkpoints for classification scripts #331

Open
d-li14 opened this issue Sep 24, 2018 · 3 comments
Open

Resuming from checkpoints for classification scripts #331

d-li14 opened this issue Sep 24, 2018 · 3 comments
Assignees
Labels
enhancement New feature or request good first issue Good for newcomers

Comments

@d-li14
Copy link

d-li14 commented Sep 24, 2018

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.

@hetong007 hetong007 self-assigned this Sep 24, 2018
@hetong007 hetong007 added enhancement New feature or request good first issue Good for newcomers labels Sep 24, 2018
@hetong007
Copy link
Member

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.

@d-li14
Copy link
Author

d-li14 commented Sep 25, 2018

OK, thanks for adding this new feature.

@jorgetony253
Copy link

Como me llega el código de liberación

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

3 participants