CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-18 20:02:25 +12:00
Code Issues Releases Wiki Activity

Fix progressbar

Browse source
This commit is contained in:
nagadomi 2016-04-16 03:45:21 +09:00
parent ea780f1871
commit a92b2cb386
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
train.lua
View file

@ -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