search

siemanko / tensorflow-deepq

A deep Q learning demonstration using Google Tensorflow
MIT License
1.17k stars 295 forks source link

where does the tensorflow model gets saved #22

Closed kumarsameer closed 8 years ago

kumarsameer commented 8 years ago

I tried to figure out tf.train.Saver or something similar in the code, but couldn't locate it. So, where does the model ( *.ckpt ) gets saved ? I can write my own saver, but since there is saved model for karpathy game already, I was wondering if I am missing something ?

siemanko commented 8 years ago

In this notebook:

https://github.com/nivwusquorum/tensorflow-deepq/blob/master/notebooks/karpathy_game.ipynb

Create a saver:

saver = tf.train.Saver(brain.variables())
saver.restore(session, "path/to/saved/model")