恢复tensorflow中的变量而不指定其形状

Question

我可以在tensorflow中这样恢复变量：

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

但如果形状未知，如何恢复它呢？当我只想加载一个变量并用它来表示performclassification时，这非常有用。在

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。在