keras.load_model () не может распознать функции активации Tensorflow

Question

Я сохранил модель tf.keras, используя функции tf.keras.save_model. почему tf.keras.load_model выдает исключение?

пример кода:

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

model = keras.Sequential([
    layers.Dense(8, activation=tf.nn.leaky_relu),
    layers.Dense(8, activation=tf.nn.leaky_relu)
])

tf.keras.models.save_model(
    model,
    'model'
)

tf.keras.models.load_model('model')

Я ожидаю, что этот код загрузит модель, но выдает исключение:

ValueError: Unknown activation function:leaky_relu

спасибо за вашу помощь!

Answer 1

Вам нужно добавить пользовательские объекты

tf.keras.models.load_model('model', custom_objects={'leaky_relu': tf.nn.leaky_relu})