TypeError: Unexpected keyword argument passed to optimizer: learning_rate
加载keras模型时,出现以上问题,代码如下
model = load_model("lstm_feature.hdf5")
主要原因:版本不匹配
解决方案(两种):
1. 训练模型可以使用
model = load_model('my_model_name.h5', custom_objects={ 'Adam': lambda **kwargs: hvd.DistributedOptimizer(keras.optimizers.Adam(**kwargs)) })
2. 如果仅仅是为了进行预测,可以直接添加compile=False
model = load_model('my_model_name.h5', compile=False)
本文解决了在加载Keras模型时遇到TypeError的问题,提供两种解决方案:一是通过自定义优化器参数,二是设置compile参数为False,适用于不同需求的场景。
1774

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



