Tensorflow使用Saver保存模型PermissionDeniedError (see above for traceback)

最新推荐文章于 2024-02-21 14:32:05 发布

徐不依

最新推荐文章于 2024-02-21 14:32:05 发布

阅读量2k

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/shulixu/article/details/90440544

在Tensorflow中，使用Saver类可以方便地保存和恢复模型参数。保存模型时，默认会保存所有参数，文件包括checkpoint、.data、.index和.meta。在恢复模型时，需指定model_filename，若文件缺少所需参数则会报错。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用tensorflow训练模型时需要保存训练后得到的模型，并在测试时加载模型。

Tensorflow中可以使用Saver类进行参数保存。

保存参数是可以选择的，如果不传参就是保存所有参数。

weight=[weights['wc1'],weights['wc2'],weights['wc3a']]
saver = tf.train.Saver(weight)#创建一个saver对象，.values是以列表的形式获取字典值
saver.save(sess,'model.ckpt')

saver_out=tf.train.Saver()
saver_out.save(sess,'file_name')

注意这里的file_name需要是绝对路径，还有人说不能和代码在一个路径里面（我没试过），另外我自己操作的时候报错PermissionDenied我以为没有写权限，然后查了一下发现是在定义路径时候文件名前必须加上“./”

checkpoint_path = os.path.join(logs_train_dir, './thing.ckpt')

运行完上面的代码之后，我们会发现在当前的程序目录下产生四个文件checkpoint、model.ckpt.data-00000-of-00001、model.ckpt.index、model.ckpt.meta。会产生四个文件的原因，之前有介