Keras ConvLSTM2D: ValueError при сохранении модели

Question

Я пытаюсь создать несколько моделей LSTM для прогнозирования временных рядов (например, Vanilla, Stacked, Bidirectional). После создания модели я хочу сохранить ее, используя tf.keras.models.save_model

Это прекрасно работает для архитектур LSTM, которые я описал выше, но при попытке сохранить ConvLSTM модель Я получаю следующую ошибку: ValueError: Словарь объектов содержит неотслеживаемый объект: (Нет, Нет) (для ключевых состояний)

Я использую Keras с бэкэндом TensorFlow (2.X) на ноутбуке Colab. Я создал блокнот , в котором можно воспроизвести проблему.

Любая помощь будет принята с благодарностью!

Редактировать: модель должна быть сохранена в формате Tensorflow SavedModel (save_format = 'tf')

Answer 1

Существует два способа сохранить модель.

1. model.save('model.h5')

2. Ваш метод, но вы пропустили название модели и расширение.

Go в каталог gdrive с использованием кд.

% cd /content/gdrive

Сохранить с именем и расширением файла.

# save model to drive
tf.keras.models.save_model(
    model = model,
    filepath = 'model2.h5',
    overwrite=True,
    include_optimizer=True,
    save_format=None,
    signatures=None
)