From 64e65ecfc91b0b8f6c1157090ceff42440a4e831 Mon Sep 17 00:00:00 2001 From: thorek1 Date: Tue, 29 Oct 2024 12:41:34 +0100 Subject: [PATCH] fix naming --- test/neural_net_solution.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/neural_net_solution.jl b/test/neural_net_solution.jl index e46cbcd6..fb54c851 100644 --- a/test/neural_net_solution.jl +++ b/test/neural_net_solution.jl @@ -65,7 +65,7 @@ for i in 1:10 end Flux.adjust!(optim, ParameterSchedulers.next!(s)) Flux.update!(optim, neural_net, grads[1]) - push!(losses, loss) # logging, outside gradient context + push!(losses, lss) # logging, outside gradient context if epoch % 10 == 0 println("Epoch: $epoch; Loss: $lss; Opt state: $(optim.layers[1].weight.rule)") end end end