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

learn()

Browse Source
This commit is contained in:
ton
2023-07-27 10:00:20 +07:00
parent a94354efb3
commit 0da983f493
3 changed files with 159 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/type.jl
View File

@@ -106,7 +106,10 @@ Base.@kwdef mutable struct kfn_1 <: knowledgeFn
on_eRec::Union{AbstractArray, Nothing} = nothing
on_eta::Union{AbstractArray, Nothing} = nothing
on_gammaPd::Union{AbstractArray, Nothing} = nothing
on_wOutChange::Union{AbstractArray, Nothing} = nothing
on_b::Union{AbstractArray, Nothing} = nothing
on_bChange::Union{AbstractArray, Nothing} = nothing
on_firingCounter::Union{AbstractArray, Nothing} = nothing
end
@@ -219,7 +222,7 @@ function kfn_1(params::Dict)
kfn.on_zt0 = zeros(1, 1, n, batch)
kfn.on_zt1 = zeros(1, 1, n, batch)
kfn.on_refractoryCounter = zeros(1, 1, n, batch)
kfn.on_refractoryDuration = ones(1, 1, n, batch) .* 1
kfn.on_refractoryDuration = ones(1, 1, n, batch) .* 0
kfn.on_alpha = ones(1, 1, n, batch) .* (exp(-kfn.on_delta / kfn.on_tau_m))
kfn.on_phi = zeros(1, 1, n, batch)
kfn.on_epsilonRec = zeros(row, col, n, batch)
@@ -227,6 +230,8 @@ function kfn_1(params::Dict)
kfn.on_eta = zeros(1, 1, n, batch)
kfn.on_gammaPd = zeros(1, 1, n, batch) .* 0.3
kfn.on_wOutChange = zeros(row, col, n, batch)
kfn.on_b = randn(1, 1, n, batch)
kfn.on_bChange = randn(1, 1, n, batch)
# subscription
w = zeros(row, col, n)