From a92b2cb386c31c1c96bf098d22a1d08b287b0afb Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sat, 16 Apr 2016 03:45:21 +0900 Subject: [PATCH] Fix progressbar --- train.lua | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.lua b/train.lua index b779d82..ded3f16 100644 --- a/train.lua +++ b/train.lua @@ -63,7 +63,6 @@ local function validate(model, criterion, data, batch_size) data[1].y:size(3)):zero() local inputs = inputs_tmp:clone():cuda() local targets = targets_tmp:clone():cuda() - for t = 1, #data, batch_size do if t + batch_size -1 > #data then break @@ -77,7 +76,7 @@ local function validate(model, criterion, data, batch_size) local z = model:forward(inputs) loss = loss + criterion:forward(z, targets) loss_count = loss_count + 1 - if t % 10 == 0 then + if loss_count % 10 == 0 then xlua.progress(t, #data) collectgarbage() end