tensorflow - 指定名称以优化为将来恢复

Question

我tensorflow创建模型和最后几行之一，它是

import tensorflow as tf 
... 
train_step = tf.train.AdagradOptimizer(LEARNING_RATE).minimize(some_loss_function) 

我不知道如果我可以给这个张/操作的名字，这样我可以在保存到磁盘后按名称恢复它？

或者，如果我不能给它一个名字，我怎么能找到它的 以下命令的输出：

tf.get_default_graph().get_operations() 

Answer 1

据​​是的，是的，你可以。

train_step = tf.train.AdamOptimizer().minimize(loss, name='my_training_step') 

可以再用以后恢复OP：

saver = tf.train.Saver(...) 
sess = tf.Session() 
saver.restore(sess, 'path/to/model') 
train_op = sess.graph.get_operation_by_name('my_training_step') 

您也可以存储在一个集合在训练操作和importing the meta graph恢复。添加到收藏和保存的样子：

saver = tf.train.Saver(...) 
tf.add_to_collection('train_step', train_step) 
# ... 
with tf.Session() as sess: 
    # ... 
    sess.save(sess, ...) 

和恢复的样子：

new_saver = tf.train.import_meta_graph('path/to/metagraph') 
new_saver.restore(sess, 'path/to/model') 
train_op = tf.get_collection('train_step')[0] # restore the op