将tensorflow的深度学习模型训练好并保存后,加载保存的模型进行测试的时候,即最后一行代码一直都报错:
#在加载模型时,所有参数赋值为其滑动平均值
ema =tf.train.ExponentialMovingAverage(Arrhythmia_backward1.MOVING_AVERAGE_DECAY)
ema_restore = ema.variables_to_restore()
saver=tf.train.Saver(ema_restore)
tensorflow.python.framework.errors_impl.NotFoundError: Key Variable not found in checkpoint
明明变量定义好了,训练也没问题,模型也已经保存在对应路径了,就是一直在报错,抓狂。前后检查并改了多次代码也没有解决,贴上试过的别的博主提供的解决方案:
1、首先是自己定义的参数变量是否和保存的参数变量的类型是一致的
2、在文件下面是否有一个叫做chockpoint的东西存在
3、最后如果运行多次出现NotFoundError (see above for traceback): Key Variable not found in checkpoint
[[Node: save_2/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT,
这种情况,请使用tf.reset