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')
然后就报错如下
原创
2021-01-01 16:32:26 ·
5335 阅读 ·
0 评论