这里有一个类似的问题: Tensorflow:如何保存/恢复模型? TLDR;您需要先使用相同的TensorFlow API命令序列重新创建模型结构,然后再使用Saver对象还原权重
这不是最理想的方法,请按照Github问题#696进行操作,以使此操作更容易
0
这里有一个类似的问题: Tensorflow:如何保存/恢复模型? TLDR;您需要先使用相同的TensorFlow API命令序列重新创建模型结构,然后再使用Saver对象还原权重
这不是最理想的方法,请按照Github问题#696进行操作,以使此操作更容易
0
我正在尝试恢复TensorFlow模型。我遵循以下示例: http : //nasdag.github.io/blog/2016/01/19/classifying-bees-with-google-tensorflow/
在示例的代码末尾,我添加了以下几行:
创建了两个文件:checkpoint和model.ckpt。
在一个新的python文件(tomas_bees_predict.py)中,我有以下代码:
但是,当我执行代码时,出现此错误:
ValueError:没有要保存的变量
有没有办法读取mode.ckpt文件并查看保存了哪些变量?或者也许有人可以根据上述示例帮助保存模型并还原模型?
编辑1:
我想我尝试运行相同的代码以重新创建模型结构,但出现错误。我认为这可能与以下事实有关:此处描述的代码未使用命名变量: http : //nasdag.github.io/blog/2016/01/19/classifying-bees-with-google-tensorflow/
所以我做了这个实验。我编写了两个版本的代码(带有和不带有命名变量)来保存模型,而代码则还原了模型。
tensor_save_named_vars.py :
tensor_save_not_named_vars.py:
tensor_restore.py:
这是我执行此代码后得到的:
因此,也许可以将原始代码(请参见上面的外部链接)修改为以下形式:
但是接下来我有一个问题:恢复weight_var和bias_var变量是否足以实现预测?我在具有GPU的功能强大的机器上进行了培训,我想将模型复制到功能不那么强大的没有GPU的计算机上以运行预测。