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

add variables

Browse Source
This commit is contained in:
ton
2023-08-10 13:41:39 +07:00
parent bbda12bae8
commit 9c57af4cd1
3 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/learn.jl
View File

@@ -16,7 +16,9 @@ function compute_paramsChange!(kfn::kfn_1, modelError, outputError)
kfn.lif_eRec,
kfn.lif_wRec,
kfn.lif_wRecChange,
kfn.on_wOut,
kfn.on_wOut,
kfn.lif_firingCounter,
kfn.lif_firingTargetFrequency,
kfn.lif_arrayProjection4d,
kfn.lif_error,
modelError,
@@ -30,7 +32,9 @@ function compute_paramsChange!(kfn::kfn_1, modelError, outputError)
kfn.alif_eRec,
kfn.alif_wRec,
kfn.alif_wRecChange,
kfn.on_wOut,
kfn.on_wOut,
kfn.alif_firingCounter,
kfn.alif_firingTargetFrequency,
kfn.alif_arrayProjection4d,
kfn.alif_error,
modelError,
@@ -59,6 +63,8 @@ function lifComputeParamsChange!( phi::CuArray,
wRec::CuArray,
wRecChange::CuArray,
wOut::CuArray,
firingCounter::CuArray,
firingTargetFrequency::CuArray,
arrayProjection4d::CuArray,
nError::CuArray,
modelError::CuArray,
@@ -81,6 +87,8 @@ function lifComputeParamsChange!( phi::CuArray,
eRec .= phi .* epsilonRec
wRecChange .+= (-eta .* nError .* eRec)
#TODO frequency regulator
# reset epsilonRec
epsilonRec .= 0
end
@@ -92,6 +100,8 @@ function alifComputeParamsChange!( phi::CuArray,
wRec::CuArray,
wRecChange::CuArray,
wOut::CuArray,
firingCounter::CuArray,
firingTargetFrequency::CuArray,
arrayProjection4d::CuArray,
nError::CuArray,
modelError::CuArray,
@@ -114,6 +124,8 @@ function alifComputeParamsChange!( phi::CuArray,
eRec .= phi .* (epsilonRec .- (beta .* epsilonRecA)) # use eq. 25
wRecChange .+= (-eta .* nError .* eRec)
#TODO frequency regulator
# reset epsilonRec
epsilonRec .= 0
epsilonRecA .= 0