Skip to content
This repository has been archived by the owner on Sep 16, 2024. It is now read-only.

How to continue training after a break? #6

Open
gaopeng-eugene opened this issue Feb 8, 2017 · 3 comments
Open

How to continue training after a break? #6

gaopeng-eugene opened this issue Feb 8, 2017 · 3 comments

Comments

@gaopeng-eugene
Copy link

For example, the training process stoped after 20K iterations. How to continue training from 20K iterations?

@DrSleep
Copy link
Owner

DrSleep commented Feb 11, 2017

You would need to run train.py --restore_from with path to your ckpt file. Note also that as default, the train.py script saves only trainable variables (thus no running mean/variance are saved). You would need to modify it to save all global variables (tf.global_variables())

@gaopeng-eugene
Copy link
Author

How to restore learning rate for Adam optimiser?

@DrSleep
Copy link
Owner

DrSleep commented Feb 21, 2017

IIRC saving tf.global_variables() and then restoring should take care of that.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants