TensorFlow之保存/恢复模型

模型保存

1 # save the specific variables
2 saver = tf.train.Saver(...variables list...)
3 
4 # this will only save trainable variables
5 saver = tf.train.Saver(tf.trainable_variables()) 
6 
7 # this will save all the variables in the graph
8 saver = tf.train.Saver()

 

https://blog.csdn.net/qiusuoxiaozi/article/details/78201796

posted on 2018-08-29 11:24  西沟冯衙内  阅读(64)  评论(0编辑  收藏  举报