From c124f8e27d23f93043fffd33526fd9da7e1994e6 Mon Sep 17 00:00:00 2001 From: thorek1 Date: Tue, 12 Nov 2024 11:44:41 +0100 Subject: [PATCH] update todo on neural net --- test/neural_net_solution.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/neural_net_solution.jl b/test/neural_net_solution.jl index 2838a3f0..8da2dd31 100644 --- a/test/neural_net_solution.jl +++ b/test/neural_net_solution.jl @@ -15,6 +15,8 @@ using Zygote using LinearAlgebra BLAS.set_num_threads(Threads.nthreads()) +# TODO: train another neural net: ergodic mean and std = neural_net(parameters) which you use then to normalise inputs for the policy function neural net + ## Settings normalise = true # use asinh and tanh at the beginning if there is no normalisation recurrent = false # the internal state needs to be reset but carries a lot of information