From 1bec436408bec5fb305efb8de4518f91e0658331 Mon Sep 17 00:00:00 2001 From: ton Date: Wed, 16 Aug 2023 15:19:39 +0700 Subject: [PATCH] minor fix --- src/learn.jl | 6 +++--- src/type.jl | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/learn.jl b/src/learn.jl index d9197da..3dcd27b 100644 --- a/src/learn.jl +++ b/src/learn.jl @@ -92,7 +92,7 @@ function lifComputeParamsChange!( timeStep::CuArray, eRec .= phi .* epsilonRec wRecChange .+= (-eta .* nError .* eRec) - #TODO frequency regulator + # frequency regulator wRecChange .+= 0.0001 .* ((firingTargetFrequency - (firingCounter./timeStep)) .* timeStep) .* eta .* eRec @@ -148,8 +148,8 @@ function alifComputeParamsChange!( timeStep::CuArray, eRec .= phi .* (epsilonRec .- (beta .* epsilonRecA)) # use eq. 25 wRecChange .+= (-eta .* nError .* eRec) - #TODO frequency regulator - wRecChange .+= 0.0001 .* ((firingTargetFrequency - (firingCounter./timeStep)) ./ timeStep) .* + # frequency regulator + wRecChange .+= 0.0001 .* ((firingTargetFrequency - (firingCounter./timeStep)) .* timeStep) .* eta .* eRec # reset epsilonRec diff --git a/src/type.jl b/src/type.jl index 43b5b48..114199e 100644 --- a/src/type.jl +++ b/src/type.jl @@ -218,7 +218,7 @@ function kfn_1(params::Dict; device=cpu) kfn.lif_firingCounter = (similar(kfn.lif_wRec) .= 0) |> device # firingTargetFrequency = desired count / total sequence length - kfn.lif_firingTargetFrequency = (similar(kfn.lif_wRec) .= 0.2) |> device + kfn.lif_firingTargetFrequency = (similar(kfn.lif_wRec) .= 0.1) |> device kfn.lif_arrayProjection4d = (similar(kfn.lif_wRec) .= 1) |> device kfn.lif_recSignal = (similar(kfn.lif_wRec) .= 0) |> device @@ -267,7 +267,7 @@ function kfn_1(params::Dict; device=cpu) kfn.alif_firingCounter = (similar(kfn.alif_wRec) .= 0) |> device # firingTargetFrequency = desired count / total sequence length - kfn.alif_firingTargetFrequency = (similar(kfn.alif_wRec) .= 0.2) |> device + kfn.alif_firingTargetFrequency = (similar(kfn.alif_wRec) .= 0.1) |> device kfn.alif_arrayProjection4d = (similar(kfn.alif_wRec) .= 1) |> device kfn.alif_recSignal = (similar(kfn.alif_wRec) .= 0) |> device