diff --git a/tests/keras_contrib/layers/test_core.py b/tests/keras_contrib/layers/test_core.py index db9b8c0..4ef20bd 100644 --- a/tests/keras_contrib/layers/test_core.py +++ b/tests/keras_contrib/layers/test_core.py @@ -32,11 +32,11 @@ def test_cosinedense(): layer_test(core.CosineDense, kwargs={'units': 3, - 'W_regularizer': regularizers.l2(0.01), - 'b_regularizer': regularizers.l1(0.01), + 'kernel_regularizer': regularizers.l2(0.01), + 'bias_regularizer': regularizers.l1(0.01), 'regularizer': regularizers.l2(0.01), - 'W_constraint': constraints.MaxNorm(1), - 'b_constraint': constraints.MaxNorm(1)}, + 'kernel_constraint': constraints.MaxNorm(1), + 'bias_constraint': constraints.MaxNorm(1)}, input_shape=(3, 2)) X = np.random.randn(1, 20)