如何更改Keras中的重量衰减值?

时间:2018-12-18 14:36:22

标签: keras

经过长时间的训练,我发现我的模型过度拟合,我想增加重量衰减。我该如何在Keras中做到这一点?

在这里,我将权重衰减传递给模型。

model.add(
    Conv2D(
        64,
        (3, 3),
        padding="same",
        activation="relu",
        kernel_regularizer=keras.regularizers.l2(weight_decay),
        kernel_initializer="he_normal",
        input_shape=(32, 32, 3),
    )
)

0 个答案:

没有答案