恢复在tensorflow变量没有指定其形状

Question

我可以恢复一个变量这种方式tensorflow：

with tf.Session() as sess: 
    var = tf.Variable([6, 7]) 
    tf.train.Saver({'name': pvdm_train}).restore(sess, 'model.ckpt')) 

但是，如果形状是未知的我怎么恢复呢？当我想加载一个变量并用它来执行分类时，这非常有用。

Answer 1

您可以通过从检查点加载元图形做到这一点：

with tf.Session() as sess: 
    saver = tf.train.import_meta_graph(
     'my-save-dir/my-model-10000.meta') 
    saver.restore(sess, 'my-save-dir/my-model-10000') 

这需要你保存模型时使用write_meta_graph=True - 这虽然默认。

你可以找到更多关于这here。