Merge pull request #661 from gheinrich/dev/torch-val-batch-size

Fix Torch validation batch size import from net desc
NVIDIA · Mar 28, 2016 · 87587ba · 87587ba
2 parents b06dfaa + a27b609
commit 87587ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/torch/main.lua b/tools/torch/main.lua
@@ -300,7 +300,7 @@ local valBatchSize
 if opt.batchSize==0 then
     local defaultBatchSize = 16
     trainBatchSize = network.trainBatchSize or defaultBatchSize
-    valBatchSize = network.validBatchSize or defaultBatchSize
+    valBatchSize = network.validationBatchSize or defaultBatchSize
 else
     trainBatchSize = opt.batchSize
     valBatchSize = opt.batchSize