python - Keras 值错误 : Unknown activation function:

Question

我使用 lambda 函数创建自定义激活函数，但当我尝试上传检查点时收到错误:

ValueError: Unknown activation function:<lambda> 

功能是:

lrelu = lambda x: tf.keras.activations.relu(x, alpha=0.2)

并像这样使用:

Conv2D(filters=96, kernel_size=(3,3),strides=(2,2),activation=lrelu)

我尝试添加 custom_objects 但没有成功:

model = load_model(filepath, custom_objects = {"lrelu": lrelu})

我知道我可以用一个额外的层替换我的函数并避免这个问题，但我想知道是否有办法做到这一点。

谢谢。

Answer 1

您将需要 Lambda 层包装器 - 下面是一个最小示例。

from keras.layers import Input, Conv2D, Lambda
from keras.models import Model
import tensorflow as tf
import numpy as np

lrelu = Lambda(lambda x: tf.keras.activations.relu(x, alpha=0.2))

ipt   = Input((4,4,3))
out   = Conv2D(3, 1, activation=lrelu)(ipt)
model = Model(ipt, out)
model.compile('adam', 'mse')

x = np.random.randn(32,4,4,3)
model.fit(x, x)

32/32 [==============================] - 6s 201ms/sample - loss: 2.1475