From f049d38c06df121ebdbc3a8b6c40d820b2e0ca13 Mon Sep 17 00:00:00 2001 From: ton Date: Sun, 17 Sep 2023 19:30:14 +0700 Subject: [PATCH] dev --- previousVersion/0.0.9/userFolderExample/mainDeving.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/previousVersion/0.0.9/userFolderExample/mainDeving.jl b/previousVersion/0.0.9/userFolderExample/mainDeving.jl index d771b8f..0e7fd70 100644 --- a/previousVersion/0.0.9/userFolderExample/mainDeving.jl +++ b/previousVersion/0.0.9/userFolderExample/mainDeving.jl @@ -422,7 +422,7 @@ function train_snn(model, trainData, validateData, labelDict::Vector) println("n $n") 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 rep in 1:10 + for rep in 1:1 stop == 3 ? break : false # 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") IronpenGPU.learn!(model, progress, device) 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) 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) 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) end