Skip to content
This repository has been archived by the owner on Jan 7, 2025. It is now read-only.

Fix Torch validation batch size import from net desc #661

Merged
merged 1 commit into from
Mar 28, 2016
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion tools/torch/main.lua
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down