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-17 19:30:14 +07:00
parent 53e85d810d
commit f049d38c06
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
previousVersion/0.0.9/userFolderExample/mainDeving.jl
View File

@@ -422,7 +422,7 @@ function train_snn(model, trainData, validateData, labelDict::Vector)
println("n $n") println("n $n")
p = Progress(n, dt=1.0) # minimum update interval: 1 second p = Progress(n, dt=1.0) # minimum update interval: 1 second
for (imgBatch, labels) in trainData # imgBatch(28, 28, 4) i.e. (row, col, batch), labels(label, batch) for (imgBatch, labels) in trainData # imgBatch(28, 28, 4) i.e. (row, col, batch), labels(label, batch)
for rep in 1:10 for rep in 1:1
stop == 3 ? break : false stop == 3 ? break : false
# prepare image into input signal (10, 2, 784, 4) i.e. (row, col, timestep, batch) # prepare image into input signal (10, 2, 784, 4) i.e. (row, col, timestep, batch)
@@ -568,13 +568,13 @@ function train_snn(model, trainData, validateData, labelDict::Vector)
println("modelname $modelname epoch $epoch rep $rep label $(labels[1]) finalAnswer __ LEARNING") println("modelname $modelname epoch $epoch rep $rep label $(labels[1]) finalAnswer __ LEARNING")
IronpenGPU.learn!(model, progress, device) IronpenGPU.learn!(model, progress, device)
elseif sum(max) == 1 && findall(max)[1] -1 == labels[1] elseif sum(max) == 1 && findall(max)[1] -1 == labels[1]
println("modelname $modelname epoch $epoch $rep label $(labels[1]) finalAnswer $finalAnswer_cpu CORRECT") println("modelname $modelname epoch $epoch rep $rep label $(labels[1]) finalAnswer $finalAnswer_cpu CORRECT")
IronpenGPU.learn!(model, progress, device) IronpenGPU.learn!(model, progress, device)
elseif sum(max) == 1 && findall(max)[1] -1 != labels[1] elseif sum(max) == 1 && findall(max)[1] -1 != labels[1]
println("modelname $modelname epoch $epoch $rep label $(labels[1]) finalAnswer $finalAnswer_cpu LEARNING") println("modelname $modelname epoch $epoch rep $rep label $(labels[1]) finalAnswer $finalAnswer_cpu LEARNING")
IronpenGPU.learn!(model, progress, device) IronpenGPU.learn!(model, progress, device)
else else
println("modelname $modelname epoch $epoch $rep label $(labels[1]) finalAnswer $finalAnswer_cpu LEARNING") println("modelname $modelname epoch $epoch rep $rep label $(labels[1]) finalAnswer $finalAnswer_cpu LEARNING")
IronpenGPU.learn!(model, progress, device) IronpenGPU.learn!(model, progress, device)
end end