TypeError: неожиданный аргумент ключевого слова передан оптимизатору: learning_rate

Question

Я пытаюсь загрузить модель Keras, обученную на виртуальной машине Azure (промо NC).Но я получаю следующую ошибку:

TypeError: Неожиданный аргумент ключевого слова передан оптимизатору: learning_rate

РЕДАКТИРОВАТЬ:

Вот фрагмент кода, которыйЯ использую для загрузки мою модель:

from keras.models import load_model
model = load_model('my_model_name.h5')

Answer 1

Использовали ли вы собственный оптимизатор?

Если это так, вы можете загрузить его следующим образом:

model = load_model('my_model_name.h5', custom_objects={
    'Adam': lambda **kwargs: hvd.DistributedOptimizer(keras.optimizers.Adam(**kwargs))
})

В качестве альтернативы вы можете загрузить модель с помощью model = load_model('my_model_name.h5', compile=False), а затем добавить оптимизатор иперекомпилируйте, но это потеряет ваши сохраненные веса.