tensorflow - 当我尝试加载使用 PReLU 作为激活函数的训练模型时出现“ValueError：未知激活函数：PReLU”？

Question

由于标题是自描述的，因此ValueError: Unknown activation function: PReLU当我尝试加载我的训练有素的 CNN 模型时出现错误，该模型将 CNN 用作PReLU卷积层和密集层的激活函数。我怎样才能使用这个模型？有解决方法吗？

ps 我正在使用 的load_model功能Keras通过提供的.h5文件加载模型。

Answer 1

当使用 PReLU 等高级激活函数时，最好将其与 add() 方法一起使用，而不是使用 'Activation' 包装它。例如

activation_function = keras.layers.advanced_activations.PReLU(init='zero', weights=None)

model = Sequential()
model.add(Dense(64, input_dim=14, init='uniform'))
model.add(activation_function)