解决tensorflow测试模型时NotFoundError错误的问题
时间:2022-11-07 09:37:54|栏目:Python代码|点击: 次
错误代码如下:
NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for xxx ……
经查资料分析,错误原因可能出在加载模型时的路径问题。我采用的加载模型方法:
with tf.Session() as sess: print("Reading checkpoints...") ckpt = tf.train.get_checkpoint_state(logs_train_dir) if ckpt and ckpt.model_checkpoint_path: global_step = ckpt.model_checkpoint_path.split('/') [-1].split('-')[-1] saver.restore(sess, ckpt.model_checkpoint_path) print('Loading success, global_step is %s' % global_step) else: print('No checkpoint file found')
在保存模型时,采用的方法为
saver = tf.train.Saver() …… …… …… if step % 1000 == 0 or (step + 1) == MAX_STEP: checkpoint_path = os.path.join(logs_train_dir, './model.ckpt') saver.save(sess, checkpoint_path, global_step=step)
注意代码块中的./model.ckpt,这是关键,原来为model.ckpt就会报错。如果在加载模型时采用直接加载模型文件的方法,则:
改之前代码:
saver.restore(sess,'model.ckpt')
改之后的代码:
saver.restore(sess,'./model.ckpt')
我的改之后是没有什么问题了,如果这种方法不能解决你的问题,再查资料解决吧
上一篇:python的列表生成式,生成器和generator对象你了解吗
栏 目:Python代码
本文标题:解决tensorflow测试模型时NotFoundError错误的问题
本文地址:http://www.codeinn.net/misctech/218374.html