Я пытаюсь сохранить и восстановить модель кодера-декодера в Учебное пособие по Tensorflow .
encoder = CNN_Encoder(embedding_dim)
...
encoder.save_weights(drive_dir + '/encoder.h5')
encoder_new = CNN_Encoder(embedding_dim)
print(encoder.layers)
print(encoder_new.layers)
encoder_new.load_weights(drive_dir + '/encoder.h5')
Это вызывает ошибку в методе load_weights
:
ValueError: You are trying to load a weight file containing 1 layers into a model with 0 layers.
Функция print
выводит это:
[<tensorflow.python.keras.layers.core.Dense object at 0x7f043d1a0c50>]
[<tensorflow.python.keras.layers.core.Dense object at 0x7f067c2555f8>]
encoder_new
уже имеет слой того же класса, что и encoder
.
Я пробовал это encoder_new.load_weights(drive_dir + '/encoder.h5', by_name = False)
,
Исключение прошло, но эта модель не работает должным образом.