@@ -52,8 +52,8 @@ function test_deep_equilibrium_network_adjoint()
52
52
for solver in DEFAULT_DEQ_SOLVERS (), jacobian_regularization in (true , false )
53
53
sensealg = SteadyStateAdjoint ()
54
54
model = DeepEquilibriumNetwork (Parallel (+ ,
55
- get_dense_layer (2 , 2 ; bias = false ),
56
- get_dense_layer (2 , 2 ; bias = false )),
55
+ get_dense_layer (2 , 2 ; use_bias = false ),
56
+ get_dense_layer (2 , 2 ; use_bias = false )),
57
57
solver;
58
58
sensealg,
59
59
jacobian_regularization,
@@ -86,8 +86,8 @@ function test_skip_deep_equilibrium_network_adjoint()
86
86
for solver in DEFAULT_DEQ_SOLVERS (), jacobian_regularization in (true , false )
87
87
sensealg = SteadyStateAdjoint ()
88
88
model = SkipDeepEquilibriumNetwork (Parallel (+ ,
89
- get_dense_layer (2 , 2 ; bias = false ),
90
- get_dense_layer (2 , 2 ; bias = false )),
89
+ get_dense_layer (2 , 2 ; use_bias = false ),
90
+ get_dense_layer (2 , 2 ; use_bias = false )),
91
91
get_dense_layer (2 , 2 ),
92
92
solver;
93
93
sensealg,
@@ -121,8 +121,8 @@ function test_skip_reg_deep_equilibrium_network_adjoint()
121
121
for solver in DEFAULT_DEQ_SOLVERS (), jacobian_regularization in (true , false )
122
122
sensealg = SteadyStateAdjoint ()
123
123
model = SkipDeepEquilibriumNetwork (Parallel (+ ,
124
- get_dense_layer (2 , 2 ; bias = false ),
125
- get_dense_layer (2 , 2 ; bias = false )),
124
+ get_dense_layer (2 , 2 ; use_bias = false ),
125
+ get_dense_layer (2 , 2 ; use_bias = false )),
126
126
nothing ,
127
127
solver;
128
128
sensealg,
0 commit comments