要保存的变量应该在字典或列表中传递

时间:2018-02-27 09:51:20

标签: python tensorflow machine-learning computer-vision training-data

第一次接受过训练数据,是的,这是一个初学者问题,我仍然不知道如何检索我的价值观。

checkpoint                             model.ckpt-708.index
events.out.tfevents.151971.name        model.ckpt-708.meta
model.ckpt-506.data-00000-of-00001     model.ckpt-809.data-00000-of-00001
model.ckpt-506.index                   model.ckpt-809.index
model.ckpt-506.meta                    model.ckpt-809.meta
model.ckpt-607.data-00000-of-00001     model.ckpt-910.data-00000-of-00001
model.ckpt-607.index                   model.ckpt-910.index
model.ckpt-607.meta                    model.ckpt-910.meta
model.ckpt-708.data-00000-of-00001

这是

的输出
ls path_to_directory

然后我手动尝试打开这些文件但是它们不会打开,所以想知道如何获取我的数据和数字。

修改: 我试图引用文档https://www.tensorflow.org/programmers_guide/saved_model

with tf.Session() as sess:
  # Restore variables from disk.
  saver = tf.train.Saver('/tmp/folder/model.ckpt-2425.meta')
  saver.restore(sess, "/tmp/model.ckpt")
  print("Model restored.")

并收到此错误TypeError:Variables to save should be passed in a dict or a list

1 个答案:

答案 0 :(得分:1)

这是一种在您的情况下恢复变量的方法: https://www.tensorflow.org/api_docs/python/tf/train/import_meta_graph

with tf.Session() as sess:
  # Restore variables from disk.
  saver = tf.train.import_meta_graph('/tmp/folder/model.ckpt-2425.meta')
  saver.restore(sess, "/tmp/model.ckpt")
  print("Model restored.")