add naming

ChrisRackauckas · ChrisRackauckas · commit 7ad050c95a16 · 2024-04-21T09:18:59.000-04:00
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)]

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ chain = Lux.Chain(`
`96`	`96`	`Lux.Dense(10 => 10, Lux.mish, use_bias = false),`
`97`	`97`	`Lux.Dense(10 => 1, use_bias = false)`
`98`	`98`	`)`
`99`		`-nn = NeuralNetworkBlock(1, 1; chain = chain, rng = StableRNG(1111))`
	`99`	`+@named nn = NeuralNetworkBlock(1, 1; chain = chain, rng = StableRNG(1111))`
`100`	`100`
`101`	`101`	`eqs = [connect(model.nn_in, nn.output)`
`102`	`102`	`connect(model.nn_out, nn.input)]`