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:29:51 +07:00
parent 7a6ca16f83
commit 2a72479f83
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/learn.jl
View File

@@ -150,7 +150,6 @@ function lifComputeParamsChange!( phi::AbstractArray,
wOut::AbstractArray, wOut::AbstractArray,
modelError::AbstractArray) modelError::AbstractArray)
d1, d2, d3, d4 = size(epsilonRec) d1, d2, d3, d4 = size(epsilonRec)
error("DEBUG -> lifComputeParamsChange! $(Dates.now())")
# Bₖⱼ in paper, sum() to get each neuron's total wOut weight # Bₖⱼ in paper, sum() to get each neuron's total wOut weight
wOutSum = reshape(sum(wOut, dims=3), (d1, :, d4)) wOutSum = reshape(sum(wOut, dims=3), (d1, :, d4))
@@ -244,9 +243,6 @@ function learn!(kfn::kfn_1)
kfn.on_wOutChange, kfn.on_wOutChange,
kfn.on_arrayProjection4d) kfn.on_arrayProjection4d)
# wOut decay
kfn.on_wOut .*= 0.0001
# wrap up learning session # wrap up learning session
if kfn.learningStage == [3] if kfn.learningStage == [3]
kfn.learningStage = [0] kfn.learningStage = [0]