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

dev

Browse Source
This commit is contained in:
ton
2023-09-19 21:43:00 +07:00
parent 821f12c86b
commit 909e27a208
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/snnUtil.jl
View File

@@ -4,6 +4,7 @@ export refractoryStatus!, addNewSynapticConn!, mergeLearnWeight!, growRepeatedP
weakenNotMatureSynapse!, pruneSynapse!, rewireSynapse! weakenNotMatureSynapse!, pruneSynapse!, rewireSynapse!
using Random, GeneralUtils using Random, GeneralUtils
using ..type
#------------------------------------------------------------------------------------------------100 #------------------------------------------------------------------------------------------------100
@@ -167,7 +168,7 @@ function rewireSynapse!(wRec::AbstractArray, neuronInactivityCounter::AbstractAr
zitCumulative::AbstractArray) zitCumulative::AbstractArray)
_,_,i3,i4 = size(wRec) _,_,i3,i4 = size(wRec)
for n in 1:i3 # neuron-by-neuron for n in 1:i3 # neuron-by-neuron
if neuronInactivityCounter[1,1,n,i4][1] < -10000 # neuron die i.e. reset all weight if neuronInactivityCounter[1,1,n,i4][1] < -100000 # neuron die i.e. reset all weight
println("neuron $n die") println("neuron $n die")
neuronInactivityCounter[:,:,n,i4] .= 0 # reset neuronInactivityCounter[:,:,n,i4] .= 0 # reset
w = random_wRec(i1,i2,1,synapseConnectionNumber) w = random_wRec(i1,i2,1,synapseConnectionNumber)