在模型的保存中我遇到过一次坑:先留一下悬念,我们先说一下怎样 保存模型和怎样导入模型:
模型的保存;代码如下:
首先需要申明一个Saver对象:
saver = tf.train.Saver
with tf.Session as sess:
saver.save(sess,'save_path')
这里save_path 就很容易出现:Parent dictionary is not exiting ; 大致意思是这样的;
这里很多种方法解决,
1:save_path 弄成:./path/filename.ckpt的形式,但是我的不成功,
2:第二种就是通过导入os 模块解决:代码如下:
cwd = os.getcwd() #获取当前文件的系统路径
save_path = './save_model/filename.ckpt'
os.path.join(cwd +save_path)
好了我们再来说说模型的取得,
saver.restore(sess,'saver_path') #saver_path 包含 文件名称;