keras保存模型 save()和save_weights()

用fit()函数训练完之后,保存模型
save()函数表示的是保存模型的结构和参数。
save_weights()函数表示的是保存模型的参数。

模型的加载方法
当使用save()保存模型时,加载使用model = load_model(‘1.h5’)
当使用save_weights()保存模型时,加载模型时需要先建立好模型model,然后再model.load_weights(2.h5)

发布了22 篇原创文章 · 获赞 3 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/love_withyou/article/details/100975494