Fix progressbar
This commit is contained in:
parent
ea780f1871
commit
a92b2cb386
|
@ -63,7 +63,6 @@ local function validate(model, criterion, data, batch_size)
|
||||||
data[1].y:size(3)):zero()
|
data[1].y:size(3)):zero()
|
||||||
local inputs = inputs_tmp:clone():cuda()
|
local inputs = inputs_tmp:clone():cuda()
|
||||||
local targets = targets_tmp:clone():cuda()
|
local targets = targets_tmp:clone():cuda()
|
||||||
|
|
||||||
for t = 1, #data, batch_size do
|
for t = 1, #data, batch_size do
|
||||||
if t + batch_size -1 > #data then
|
if t + batch_size -1 > #data then
|
||||||
break
|
break
|
||||||
|
@ -77,7 +76,7 @@ local function validate(model, criterion, data, batch_size)
|
||||||
local z = model:forward(inputs)
|
local z = model:forward(inputs)
|
||||||
loss = loss + criterion:forward(z, targets)
|
loss = loss + criterion:forward(z, targets)
|
||||||
loss_count = loss_count + 1
|
loss_count = loss_count + 1
|
||||||
if t % 10 == 0 then
|
if loss_count % 10 == 0 then
|
||||||
xlua.progress(t, #data)
|
xlua.progress(t, #data)
|
||||||
collectgarbage()
|
collectgarbage()
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue