add naming

SciML · Apr 21, 2024 · 7ad050c · 7ad050c
1 parent cbb2c60
commit 7ad050c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/src/friction.md b/docs/src/friction.md
@@ -96,7 +96,7 @@ chain = Lux.Chain(
     Lux.Dense(10 => 10, Lux.mish, use_bias = false),
     Lux.Dense(10 => 1, use_bias = false)
 )
-nn = NeuralNetworkBlock(1, 1; chain = chain, rng = StableRNG(1111))
+@named nn = NeuralNetworkBlock(1, 1; chain = chain, rng = StableRNG(1111))
 
 eqs = [connect(model.nn_in, nn.output)
        connect(model.nn_out, nn.input)]

diff --git a/test/lotka_volterra.jl b/test/lotka_volterra.jl
@@ -44,7 +44,7 @@ end
 model = lotka_ude()
 
 chain = multi_layer_feed_forward(2, 2)
-nn = NeuralNetworkBlock(2, 2; chain, rng = StableRNG(42))
+@named nn = NeuralNetworkBlock(2, 2; chain, rng = StableRNG(42))
 
 eqs = [connect(model.nn_in, nn.output)
        connect(model.nn_out, nn.input)]