Загрузка пользовательской модели с Tensorflow 2.1

Question

Я создал свой собственный подкласс класса tf.keras.Model и пытаюсь сохранить и загрузить обученный экземпляр этого класса. Я пытаюсь выполнить этот учебник, но каждый раз, когда я go загружаю сохраненную модель, я получаю одно и то же сообщение об ошибке: TypeError: __init__() got an unexpected keyword argument 'reduction'. Я попытался добавить этот ключевой аргумент в мой класс, но он ничего не меняет. Есть идеи?

Answer 1

У меня была такая же проблема для Tensorflow 1.14 , и я решил ее, добавив compile=False к функции загрузки:

new_model = tf.keras.models.load_model('saved_model/my_model', compile=False)

Даже при compile=False можно запустить model.predict() function.

Решение возникло из этой проблемы Tensorflow .