在张量流中加载现有模型时出现问题

时间:2019-03-06 22:18:53

标签: python tensorflow

我在另一台机器上生成一个张量流模型,并且我的模型文件夹包含以下内容:

-model.meta
-checkpoint
-model.index
-model.data-00000-of-00001

将其复制到我的工作机后,我使用以下方法加载模型:

model1 = tf.train.import_meta_graph("models/model_epoch100_emb100_bs32_lr0.025/.meta")
model1.restore(sess, tf.train.latest_checkpoint("models/model_epoch100_emb100_bs32_lr0.025/"))

该方法曾经用于我拥有的另一个旧模型。但是现在我得到这个错误:

ERROR:tensorflow:Couldn't match files for checkpoint models/model_epoch100_emb100_bs32_lr0.025/.
ERROR:tensorflow:Couldn't match files for checkpoint models/model_epoch100_emb100_bs32_lr0.025/.
Traceback (most recent call last):
  File "/datadisk/Workspace/project/fgsm.py", line 148, in <module>
    main(config)
  File "/datadisk/Workspace/project/fgsm.py", line 76, in main
    model1.restore(sess, tf.train.latest_checkpoint("models/model_epoch100_emb100_bs32_lr0.025/"))
  File "/datadisk/Workspace/project/venv/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1534, in restore
    raise ValueError("Can't load save_path when it is None.")
ValueError: Can't load save_path when it is None.

我确保该路径确实存在,并且在intellij环境中将文件夹标记为源。我不知所措。

编辑:

我最初使用以下方式存储模型:

        saver = tf.train.Saver()
        save_path = saver.save(sess=sess, save_path="g2v_model/model")

但是我将文件夹和scp的内容复制到了上述文件夹中的本地计算机中。

Edit2:

我不知道这有多重要,但还有一件事。我生成模型的代码是使用python2,而现在使用的代码是python 3。

0 个答案:

没有答案