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/'效果一样)

1万+

被折叠的 条评论
为什么被折叠?



