本文介绍在Keras中如何使用load_model加载包含自定义层或损失函数的模型。通过提供一个示例，展示如何正确地在加载模型时指定自定义组件，确保模型能够正确读取。
在Keras中，如果存在自定义layer或者loss，需要在load_model()中以字典形式指定layer或loss。
model = load_model('1.h5', custom_objects={'my_loss':my_loss,'NestedLSTM': NestedLSTM})
https://blog.youkuaiyun.com/liushuijingying2/article/details/87695593
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
