minor fix

2023-08-15 12:00:54 +07:00
parent 19735c3df0
commit 1775684e38
3 changed files with 26 additions and 11 deletions
--- a/previousVersion/0.0.3/main.jl
+++ b/previousVersion/0.0.3/main.jl
@@ -593,7 +593,7 @@ function validate(model, dataset, labelDict)
    thinkingPeriod = 16    # 1000-784 = 216
    predict = [0]   |> device

-    n = length(trainData)
+    n = length(dataset)
    println("n $n")
    p = Progress(n, dt=1.0)   # minimum update interval: 1 second
    for (imgBatch, labels) in dataset