CODEWITHSUNDEEP
tensorflowpre-trained-model
HeikoG
HeikoG

Reputation: 1813

tensorflow retrain model file

im getting started with tensorflow und using retrain.py to teach it some new categories - this works well - however i have some questions:

In the comments of retrain.py it says: "This produces a new model file that can be loaded and run by any TensorFlow program, for example the label_image sample code" however I havent found where this new model file is saved to ?

also: it does contain the whole model, right ? not just the retrained part ?

Thanks for clearing this up

Upvotes: 0

Views: 570

Answers (2)

sdr2002
sdr2002

Reputation: 552

1)I think you may want to save the new model. When you want to save a model after some process, you can use saver.save(sess, 'directory/model-name', *optional-arg).

Check out https://www.tensorflow.org/api_docs/python/tf/train/Saver

If you change model-name by epoch or any measure you would like to use, you can save the new model(otherwise, it may overlap with previous models saved).

You can find the model saved by searching 'checkpoint', '.index', '.meta'.

2)Saving the whole model or just part of it? It's the part you need to learn bunch of ideas on tf.session and savers. You can save either the whole or just part, it's up to you. Again, start from the above link. The moral is that you put the variables you would like to save in a list quoted as 'var_list' in the link, and you can save only for them. When you call them back, you now also need to specify which variables in your model correspond to the variables in the loaded variables.

Upvotes: 1

Shrikant Shete
Shrikant Shete

Reputation: 329

While running retrain.py you can give --output_graph and --output_labels parameters which specify the location to save graph (default is /tmp/output_graph.pb) and the labels as well. You can change those as per your requirements.

Upvotes: 0

Related Questions