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的更多信息。

在张量流中恢复变量而不指定其形状

1 个答案: