Can I stop train.py and resume it later?

hunkim / word-rnn-tensorflow

Multi-layer Recurrent Neural Networks (LSTM, RNN) for word-level language models in Python using TensorFlow.

MIT License

1.3k stars 494 forks source link

Can I stop train.py and resume it later? #60

Closed ghost closed 7 years ago

ghost commented 7 years ago

Pretty much what it says in the title. I've noticed that it occasionally saves the model, so I was wondering if it's safe for me to stop it and resume it later.

normanheckscher commented 7 years ago

When resuming you'll need to use the '--init_from' flag to set the location of the saved files which are to be resumed.

normanheckscher commented 7 years ago

python train.py --init_from=/Users/kartikmokashi/Desktop/word-rnn-tensorflow-master/save/

ghost commented 7 years ago

Oh thanks I just tried that and it works