version 0.0.2

src/forward.jl

@@ -18,7 +18,19 @@ function (kfn::kfn_1)(input::AbstractArray)
     #TODO time step forward
     if view(kfn.learningStage, 1)[1] == 1
         # reset learning params
-        # kfn.learningStage = [2]
+        kfn.lif_vt .= 0
+        kfn.lif_wRecChange .= 0
+        kfn.lif_epsilonRec .= 0
+        
+        kfn.alif_vt .= 0
+        kfn.alif_epsilonRec .= 0
+        kfn.alif_wRecChange .= 0
+
+        kfn.on_vt .= 0
+        kfn.on_epsilonRec .= 0
+        kfn.on_wOutChange .= 0
+
+        kfn.learningStage = [2]
     end
     
     # update activation matrix with "lif_zt1" and "alif_zt1" by concatenating
@@ -27,12 +39,6 @@ function (kfn::kfn_1)(input::AbstractArray)
                 reshape(kfn.lif_zt, (size(input, 1), :, 1, size(input, 3))), 
                 reshape(kfn.alif_zt, (size(input, 1), :, 1, size(input, 3))), dims=2)
     kfn.zit .= reshape(_zit, (size(input, 1), :, size(input, 3)))
-    
-    # pass input_data into input neuron.
-#     GeneralUtils.cartesianAssign!(kfn.zit, input)
-
-#     kfn.zit = kfn.zit |> device
-#     input = input |> device
 
     # project 3D kfn zit into 4D lif zit
     i1, i2, i3, i4 = size(kfn.lif_zit)