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)