如何从张量流中保存的检查点恢复特定范围的变量？

import tensorflow as tf 
saver = tf.train.Saver() 
saver.restore(...)

但是saver.restore只有恢复整个图的选项。我想只恢复那些在特定范围内的变量。如何从张量流中保存的检查点恢复特定范围的变量？

在此先感谢！

来源

2017-03-02 Prasanna Parthasarathi

假设您已经在范围InceptionV1范围内有Google的InceptionNet模型，并且您想要加载它，但要重新训练的范围InceptionRetrained的最后一层除外。

假设你已经开始重新训练的最后一层，你通过saver2.save(session, 'last_layer.ckpt')创建last_layer.ckpt文件，这里是如何从检查点恢复净。

saver1 = tf.train.Saver(var_list=tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='InceptionV1')) 
saver1.restore(session, 'inception_model_from_google.ckpt') 

saver2 = tf.train.Saver(var_list=tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='InceptionRetrained')) 
saver2.restore(session, 'last_layer.ckpt')

如果你再培训只剩下最后一层，不要忘记通过调用优化与var_list参数禁用梯度向上的网络（节省时间）的传播。

tf.train.Optimizer(0.0001).minimize(
      loss, var_list=tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='Inceptionretrained'))

来源

2017-03-23 13:29:45 user2781994

如何从张量流中保存的检查点恢复特定范围的变量？

回答

相关问题