tfutils icon indicating copy to clipboard operation
tfutils copied to clipboard

Write out graph definition when storing weights for restoration from graph definition and Tensorboard visualization

Open damro opened this issue 7 years ago • 2 comments

For debugging as well as for book keeping purposes, it would be useful to write out the complete graph definition every time weights are saved.

We can store out the graph definition using tf.train.write_graph() as a ".pbtxt" file, (pickle it) and store it in gridfs:

https://www.tensorflow.org/versions/master/api_docs/python/tf/train/write_graph

This description can be then loaded into Tensorboard or used to rebuild the model without having to keep the actual script around / to make sure the correct script is being used to restart training.

damro avatar Oct 27 '17 19:10 damro

I like this, it might also be good to save a .png of the structure too just for visualization purposes

anayebi avatar Oct 27 '17 20:10 anayebi

See also #28

chengxuz avatar Sep 20 '18 23:09 chengxuz