Merge pull request #91 from SciML/lux_upgrade

Upgrade bias -> use_bias
SciML · Aug 1, 2023 · 3ffec08 · 3ffec08
2 parents 2e9c500 + 5ffb5c1
commit 3ffec08
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -35,8 +35,8 @@ Random.seed!(rng, seed)
 
 model = Lux.Chain(Lux.Dense(2, 2),
     DEQs.DeepEquilibriumNetwork(Lux.Parallel(+,
-            Lux.Dense(2, 2; bias=false),
-            Lux.Dense(2, 2; bias=false)),
+            Lux.Dense(2, 2; use_bias=false),
+            Lux.Dense(2, 2; use_bias=false)),
         DEQs.ContinuousDEQSolver(;
             abstol=0.1f0,
             reltol=0.1f0,

diff --git a/docs/src/index.md b/docs/src/index.md
@@ -28,8 +28,8 @@ Random.seed!(rng, seed)
 
 model = Lux.Chain(Lux.Dense(2, 2),
     DEQs.DeepEquilibriumNetwork(Lux.Parallel(+,
-            Lux.Dense(2, 2; bias=false),
-            Lux.Dense(2, 2; bias=false)),
+            Lux.Dense(2, 2; use_bias=false),
+            Lux.Dense(2, 2; use_bias=false)),
         DEQs.ContinuousDEQSolver(;
             abstol=0.1f0,
             reltol=0.1f0,

diff --git a/test/adjoint.jl b/test/adjoint.jl
@@ -52,8 +52,8 @@ function test_deep_equilibrium_network_adjoint()
     for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
         sensealg = SteadyStateAdjoint()
         model = DeepEquilibriumNetwork(Parallel(+,
-                get_dense_layer(2, 2; bias=false),
-                get_dense_layer(2, 2; bias=false)),
+                get_dense_layer(2, 2; use_bias=false),
+                get_dense_layer(2, 2; use_bias=false)),
             solver;
             sensealg,
             jacobian_regularization,
@@ -86,8 +86,8 @@ function test_skip_deep_equilibrium_network_adjoint()
     for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
         sensealg = SteadyStateAdjoint()
         model = SkipDeepEquilibriumNetwork(Parallel(+,
-                get_dense_layer(2, 2; bias=false),
-                get_dense_layer(2, 2; bias=false)),
+                get_dense_layer(2, 2; use_bias=false),
+                get_dense_layer(2, 2; use_bias=false)),
             get_dense_layer(2, 2),
             solver;
             sensealg,
@@ -121,8 +121,8 @@ function test_skip_reg_deep_equilibrium_network_adjoint()
     for solver in DEFAULT_DEQ_SOLVERS(), jacobian_regularization in (true, false)
         sensealg = SteadyStateAdjoint()
         model = SkipDeepEquilibriumNetwork(Parallel(+,
-                get_dense_layer(2, 2; bias=false),
-                get_dense_layer(2, 2; bias=false)),
+                get_dense_layer(2, 2; use_bias=false),
+                get_dense_layer(2, 2; use_bias=false)),
             nothing,
             solver;
             sensealg,