在加载的图形中替换迭代器

时间:2018-10-15 10:52:10

标签: python tensorflow

我正在从检查点文件中恢复图形。使用数据集迭代器为网络提供数据。我想用一个新的迭代器替换该迭代器,因为我想评估一些新示例的结果。

我找不到用新的替代旧迭代器的有效方法。

这是我尝试过的:

saver = tf.train.import_meta_graph("{}.meta".format(checkpoint_file), input_map={})
iterator = tf.get_default_graph().get_operation_by_name("DataPipe/Iterator")
init_op = iterator.make_initializer(dataset)

但是出现以下错误:

  

AttributeError:“ Operation”对象没有属性“ make_initializer”

我该怎么办?优先考虑重用我已经训练的网络。

编辑:

这是related question,但是我的网络已经受过训练,所以我现在不能更改其定义。

0 个答案:

没有答案