mirror of
https://github.com/wassname/keras-contrib.git
synced 2026-08-14 12:30:09 +08:00
fix cos dense test
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user