在Tensorflow v1.14中还原保存的模型

学习者

我正在使用tensorflow v1.14。我有一个已保存的模型,并且正在尝试使用以下代码还原该模型:

loader = tf.train.import_meta_graph("models/fcnn0/model.ckpt.meta")
graph = tf.get_default_graph()
sess = tf.Session()
loader.restore(sess, "models/fcnn0/model.ckpt")

我曾经在Tensorflow v1.13中使用同一段代码,并且过去可以正常工作。但是现在我得到了错误

Traceback (most recent call last):
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1356, in _do_call
    return fn(*args)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1341, in _run_fn
    options, feed_dict, fetch_list, target_list, run_metadata)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1429, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.DataLossError: file is too short to be an sstable
     [[{{node save/RestoreV2}}]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/sandesh/PycharmProjects/fading/finding_code/src/load_32_64.py", line 8, in <module>
    loader.restore(sess, "models/fcnn_32_64_aenc_1331_747_3870000/model.ckpt")
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1286, in restore
    {self.saver_def.filename_tensor_name: save_path})
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 950, in run
    run_metadata_ptr)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1173, in _run
    feed_dict_tensor, options, run_metadata)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1350, in _do_run
    run_metadata)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1370, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.DataLossError: file is too short to be an sstable
     [[node save/RestoreV2 (defined at home/sandesh/PycharmProjects/fading/finding_code/src/load_32_64.py:5) ]]

Original stack trace for 'save/RestoreV2':
  File "home/sandesh/PycharmProjects/fading/finding_code/src/load_32_64.py", line 5, in <module>
    loader = tf.train.import_meta_graph("models/fcnn_32_64_aenc0/model.ckpt.meta")
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1449, in import_meta_graph
    **kwargs)[0]
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1473, in _import_meta_graph_with_return_elements
    **kwargs))
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/meta_graph.py", line 857, in import_scoped_meta_graph_with_return_elements
    return_elements=return_elements)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/importer.py", line 443, in import_graph_def
    _ProcessNewOps(graph)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/importer.py", line 236, in _ProcessNewOps
    for new_op in graph._add_new_tf_operations(compute_devices=False):  # pylint: disable=protected-access
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 3751, in _add_new_tf_operations
    for c_op in c_api_util.new_tf_operations(self)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 3751, in <listcomp>
    for c_op in c_api_util.new_tf_operations(self)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 3641, in _create_op_from_tf_operation
    ret = Operation(c_op, self)
  File "usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 2005, in __init__
    self._traceback = tf_stack.extract_stack()

有人可以指出我做错了什么吗?提前致谢。

学习者

我正在查看保存模型文件的文件夹,发现该model.ckpt.meta文件未正确写入磁盘。我重新运行训练并保存了模型,然后它运行完美。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

无法使用更新代码在 tensorflow V2 中加载 Tensorflow v1 模型?

在TensorFlow v1中隐式启用TensorFlow v2行为

如何在v1.13.1中还原tensorflow v1.1.0保存的模型

TensorFlow v1和v2中Keras的ZeroPadding2D的区别?

如何在TensorFlow v1中使用freeze_graph.py工具

无法在Tensorflow V0.8中还原模型

在TensorFlow中还原模型

删除端点网址中的v1

AttributeError:模块“ tensorflow.compat”没有属性“ v1” Tensorflow v:1.10.0

在Tensorflow中还原保存的神经网络

TensorFlow:模型保存成功但还原失败,我在哪里错?

tensorflow 0.9 skflow模型保存和还原不起作用

Tensorflow v1 名称“features:0”指的是一个不存在的 Tensor

_api / v1 / hard /和python / hard之间的tensorflow / hard utils模块混淆

使用tf.control_dependencies(tensorflow v1)的意外(随机)执行顺序

Api :: V1 :: UsersController#create中的ActionController InvalidAuthenticityToken

版本“ v1”中的部署不能作为部署处理:

使用angularjs在ionic v1中使用querystring

Ionic v1中的img src路径

Google 地图在 Ionic v1 中无法正常工作

React-Bootstrap v1 中的 Panel.Collapse

如何在Ionic V1中使用FCM

交易何时在 HLF v1 中完成?

空手道 v1 中的表问题

已保存模型的Tensorflow服务ssd_mobilenet_v1_coco

Tensorflow:如何使用还原的模型?

Tensorflow模型占位符还原

如何在 tensorflow lite 模型文件 (v 1.x) 中嵌入模型版本或描述

将 cntk 模型 v1 转换为 cntk 模型 v2