Tensorflow最新的版本,在使用Saver载入模型参数时会报错:
Tensorflow :Unsuccessful TensorSliceReader constructor: Failed to find any matching files
意思是说找不到模型保存的文件。
在Saver.save中我们传入sess,modelpath
例如:
dir=“D:\data\Iris\”
saver.save(sess,dir+“my-model.cpk”)
然后保存后,我们发现目录下的文件为:
D:\data\Iris\
my-model.cpk.data-0000-of-00001 #变量的键值对
my-model.cpk.index
my-model.cpk.meta #模型的拓扑结构
都是有后缀的,也的确找不到my-model.cpk,根据github所说得这是tensorflow新版本的bug。
解决方法:
saver.restore(sess,dir+".\\"+“my-model.cpk”)
即可