无法还原Tensorflow检查点

时间:2019-03-25 10:54:57

标签: tensorflow

我正在尝试在Tensorflow中恢复已保存的检查点。该模型在this链接中可用。但是,我找不到任何还原变量。我不明白我在做什么错。

我尝试了以下代码。

PATH_REL_META = r'./model.ckpt.meta'

# start tensorflow session
with tf.Session() as sess:

    # import graph
    saver = tf.train.import_meta_graph(PATH_REL_META)

    # load weights for graph
    saver.restore(sess, PATH_REL_META[:-5])

    # get all global variables (including model variables)
    vars_global = tf.global_variables()

    # get their name and value and put them into dictionary
    sess.as_default()
    model_vars = {}
    for var in vars_global:
        try:
            model_vars[var.name] = var.eval()
        except:
            print("For var={}, an exception occurred".format(var.name))
    print (model_vars)

model_vars 变量在我打印时只是一个空数组。为什么我无法正确还原检查点?

0 个答案:

没有答案
相关问题