KeyError:恢复张量流模型时出现“ IteratorV2”

时间:2019-04-02 09:00:10

标签: python tensorflow jupyter google-colaboratory

我已经在使用gnu加速运行tensorflow 1.13.1的google colab上训练了seq2seq模型,然后尝试在jupyter的cpu上运行tensorflow 1.9.0的本地计算机上恢复该模型。 当我恢复模型图时:

loader = tf.train.import_meta_graph(model_graph_path)

我得到了错误:

KeyError: 'IteratorV2'.

由于不同的tensorflow版本而引起的问题吗?因为如果我完全在同一台机器上(无论是衣领vm还是本地机器)训练并恢复我的模型,模型都可以训练并恢复正常。 如果需要更多代码来理解可能的问题,请随时提出。 预先谢谢你。

0 个答案:

没有答案
相关问题