Я пытаюсь сохранить модель тензор потока 2, которая содержит слой внимания (tf.keras.layers.attention).
Я пытался использовать model.save () и model.to_ json (), оба выдают следующую ошибку:
NotImplementedError: Слои с аргументами в __init__ должны переопределить get_config.
Нет ошибки при удалении слоя внимания из модели.
Я должен уточнить, что я использую Keras в Tensorflow 2 для обработки сессий и слоев модели / компиляции / обучения. Я могу сохранить обученную модель керас для повторяющейся сети без внимания. Однако, когда я использую keras.layers.Attention, модель все еще может компилироваться и обучаться, но не будет сохранять с помощью приведенных выше команд.