问:Tensorflow:如何保存/恢复tf.data.Dataset?

时间:2017-11-29 14:02:19

标签: tensorflow save dataset restore

我用tf.data.Dataset()作为数据IO函数

创建了一个模型

然后我导出图并尝试使用meta_graph文件恢复它 但它失败了并且发生了错误消息。

我认为tf.data.Dataset()创建了一个C ++对象而不是之前使用过的python队列。

graph_def只有一个C ++对象处理程序引用,所以单独graph_def没有真正的C ++对象就无法加载完整的图形。

如何使用tf.data.Dataset()加载可执行图? 或者现在不可能吗?

File "/usr/local/lib/python3.4/dist-packages/tensorflow/python/framework/ops.py", line 1470, in __init__
    self._traceback = self._graph._extract_stack()  # pylint: disable=protected-access

NotFoundError (see above for traceback): Function
_make_dataset_5150cb86 is not defined.
         [[Node: batch_processing/OneShotIterator = OneShotIterator[container="", dataset_factory=_make_dataset_5150cb86[], output_shapes=[[?,1], [?,299,299,3]], output_types=[DT_INT32, DT_FLOAT], shared_name="",
_device="/job:workers/replica:0/task:0/device:CPU:0"]()]]

0 个答案:

没有答案