From 160128d7a8d11693856c23959f2c504ca07c3068 Mon Sep 17 00:00:00 2001
From: ton <narawat@gmail.com>
Date: Thu, 10 Aug 2023 13:31:02 +0700
Subject: [PATCH] minor fix

---
 src/learn.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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