diff --git a/src/learn.jl b/src/learn.jl index 9f185d1..f147b2a 100644 --- a/src/learn.jl +++ b/src/learn.jl @@ -279,7 +279,7 @@ function onLearn!(wOut, wOutChange, arrayProjection4d) # merge learning weight with average learning weight - wRec .+= (sum(wRecChange, dims=4) ./ (size(wRec, 4))) .* arrayProjection4d + wOut .+= (sum(wOutChange, dims=4) ./ (size(wOut, 4))) .* arrayProjection4d # adaptive wOut to help convergence using c_decay wOut .-= 0.001 .* wOut