keras.load_model() 无法识别 Tensorflow 的激活函数

Question

我使用 tf.keras.save_model 函数保存了一个 tf.keras 模型。为什么 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})

keras.load_model() 无法识别 Tensorflow 的激活函数

keras.load_model() can't recognize Tensorflow's activation functions

tensorflow

keras

tf.keras