tensorflow中模型保存:tf.train.Saver.save,tf.train.train.Saver.restore

在模型的保存中我遇到过一次坑:先留一下悬念,我们先说一下怎样 保存模型和怎样导入模型:

模型的保存;代码如下:

首先需要申明一个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 包含 文件名称;

猜你喜欢

转载自blog.csdn.net/weixin_38859557/article/details/80878013