keras模型保存HDF5和H5出现的问题(ValueError: Parent directory of xxx.hdf5 doesn‘t exist, can‘t save)

keras版本以及tensorflow-gpu 版本:

Name: Keras
Version: 2.1.3

Name: tensorflow-gpu
Version: 1.10.0

问题重现

假设model是一个已经训练好的模型,现在要用keras中自带的save_weights保存模型的权重。一般我们都直接保存为‘xxx.h5’(save_weights只会保存权重)。我尝试用下面一行代码将其保存下来

model.save_weights('cifar10.hdf5')

然后就报错如下

ValueError: Parent directory of cifar10.hdf5 doesn't exist, can't save

但是如果更改命名的后缀为‘h5’,就可以顺利保存

model.save_weights('cifar10.h5')

出现在文件夹保存成功,开始想这是为什么,寻找相关的资料大多数都说hdf5文件和h5文件是同一个东西。

解决方案

因为这里我用到了keras的回调函数(callbacks)来保存最佳的模型,检查报错原因,可能是因为没有有效的父目录。于是使用如下代码:(使用绝对路径,使用'/python/kerastest/cifar10/'效果一样)


                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值