fix cos dense test

This commit is contained in:
farizrahman4u
2017-03-19 09:31:37 +05:30
parent 8c2eac4853
commit 8fa0c2dbc1
+4 -4
View File
@@ -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)