1.现象说明
tensorflow恢复训练好的参数进行推理时出现以下报错(图1所示)
图1
2.原因分析
根据报错提示,是在创建saver对象时出错,具体错误是因为“没有变量可以存储”,现在让我们看一下具体的代码,如图2所示。为了说明问题,代码很简单,原因就是在创建saver时,当前计算图中没有任何变量,因此会爆出以上错误。
图2
3.解决办法
解决办法很简单,就是推理时,先将模型结构加载到计算图(简单来说就是需要先定义变量),然后创建saver对象即可。改成图3所示,就不会报错了。
图3