ton/IronpenGPU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
ton
2023-08-10 13:28:33 +07:00
parent a80e9f2621
commit 7a6ca16f83
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/learn.jl
View File

@@ -79,7 +79,7 @@ function lifComputeParamsChange!( phi::CuArray,
# this neuron receive summed error signal (modelError) # this neuron receive summed error signal (modelError)
nError .= (modelError .* wOutSum) .* arrayProjection4d nError .= (modelError .* wOutSum) .* arrayProjection4d
eRec .= phi .* epsilonRec eRec .= phi .* epsilonRec
wRecChange .+= ((-1 .* eta) .* nError .* eRec) wRecChange .+= (-eta .* nError .* eRec)
# reset epsilonRec # reset epsilonRec
epsilonRec .= 0 epsilonRec .= 0
@@ -112,7 +112,7 @@ function alifComputeParamsChange!( phi::CuArray,
# this neuron receive summed error signal (modelError) # this neuron receive summed error signal (modelError)
nError .= (modelError .* wOutSum) .* arrayProjection4d nError .= (modelError .* wOutSum) .* arrayProjection4d
eRec .= phi .* (epsilonRec .- (beta .* epsilonRecA)) # use eq. 25 eRec .= phi .* (epsilonRec .- (beta .* epsilonRecA)) # use eq. 25
wRecChange .+= ((-1 .* eta) .* nError .* eRec) wRecChange .+= (-eta .* nError .* eRec)
# reset epsilonRec # reset epsilonRec
epsilonRec .= 0 epsilonRec .= 0
@@ -134,7 +134,7 @@ function onComputeParamsChange!(phi::CuArray,
eRec .= phi .* epsilonRec eRec .= phi .* epsilonRec
nError .= reshape(outputError, (1, 1, :, size(outputError, 2))) .* arrayProjection4d nError .= reshape(outputError, (1, 1, :, size(outputError, 2))) .* arrayProjection4d
wOutChange .+= ((-1 .* eta) .* nError .* eRec) wOutChange .+= (-eta .* nError .* eRec)
# reset epsilonRec # reset epsilonRec
epsilonRec .= 0 epsilonRec .= 0