Skip to content

Commit 5ffb5c1

Browse files
Upgrade bias -> use_bias
1 parent 2e9c500 commit 5ffb5c1

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ Random.seed!(rng, seed)
3535

3636
model = Lux.Chain(Lux.Dense(2, 2),
3737
DEQs.DeepEquilibriumNetwork(Lux.Parallel(+,
38-
Lux.Dense(2, 2; bias=false),
39-
Lux.Dense(2, 2; bias=false)),
38+
Lux.Dense(2, 2; use_bias=false),
39+
Lux.Dense(2, 2; use_bias=false)),
4040
DEQs.ContinuousDEQSolver(;
4141
abstol=0.1f0,
4242
reltol=0.1f0,

docs/src/index.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ Random.seed!(rng, seed)
2828

2929
model = Lux.Chain(Lux.Dense(2, 2),
3030
DEQs.DeepEquilibriumNetwork(Lux.Parallel(+,
31-
Lux.Dense(2, 2; bias=false),
32-
Lux.Dense(2, 2; bias=false)),
31+
Lux.Dense(2, 2; use_bias=false),
32+
Lux.Dense(2, 2; use_bias=false)),
3333
DEQs.ContinuousDEQSolver(;
3434
abstol=0.1f0,
3535
reltol=0.1f0,

test/adjoint.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ function test_deep_equilibrium_network_adjoint()
5252
for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
5353
sensealg = SteadyStateAdjoint()
5454
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)),
5757
solver;
5858
sensealg,
5959
jacobian_regularization,
@@ -86,8 +86,8 @@ function test_skip_deep_equilibrium_network_adjoint()
8686
for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
8787
sensealg = SteadyStateAdjoint()
8888
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)),
9191
get_dense_layer(2, 2),
9292
solver;
9393
sensealg,
@@ -121,8 +121,8 @@ function test_skip_reg_deep_equilibrium_network_adjoint()
121121
for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
122122
sensealg = SteadyStateAdjoint()
123123
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)),
126126
nothing,
127127
solver;
128128
sensealg,

0 commit comments

Comments
 (0)