Example on how to freeze the model

martin-gorner / tensorflow-rnn-shakespeare

Code from the "Tensorflow and deep learning - without a PhD, Part 2" session on Recurrent Neural Networks.

Apache License 2.0

534 stars 249 forks source link

Example on how to freeze the model #18

Closed ghost closed 7 years ago

ghost commented 7 years ago

Can you provide an example on how to freeze the shakespeare-rnn? It would be nice to save the model into a .pb file.

martin-gorner commented 7 years ago

rnn_train.py saves the model on a regular basis during training. You will find the model files in a directory named "checkpoints". A bunch of those files are also available for download (see README). You can play them back using rnn_play.py

ghost commented 6 years ago

Thanks Martin, but I did not see any code that converts the model to a single .pb file. Do you know of any examples that show an rnn model get saved into a single .pb file?