Skip to content
New issue

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

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

Already on GitHub? # to your account

Load a model and resume training? #12

Open
smahajan07 opened this issue May 27, 2017 · 2 comments
Open

Load a model and resume training? #12

smahajan07 opened this issue May 27, 2017 · 2 comments

Comments

@smahajan07
Copy link

Is there a way to load the model and resume training? I am training on approx 75k images and it is taking almost 2 days (GTX 1080). So is there any way to resume training from the last checkpoint?

@nilboy
Copy link
Owner

nilboy commented May 27, 2017

https://github.com/nilboy/pixel-recursive-super-resolution/blob/master/solver.py#L55
saver.restore(sess, './models/model.ckpt-30000')

@smahajan07
Copy link
Author

@nilboy I am trying to load the last checkpoint files I have, in a replica file of solver.py and commented out the portion for loss, since I don't need backprop now and only calling the 'sample' function to save the image as per the session loaded.

Is this the right way to create an inference?
If you have time, please guide me on how to create an inference. Or best, if you can upload a file for testing the trained models.

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants