CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

merge

Browse source
This commit is contained in:
nagadomi 2017-02-23 08:11:38 +09:00
parent a121eb39cb dac1b89750
commit 39c7ce1ea7
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
train.lua
View file

@ -290,7 +290,7 @@ local function validate(model, criterion, eval_metric, data, batch_size)
local batch_mse = eval_metric:forward(z, targets)
loss = loss + criterion:forward(z, targets)
mse = mse + batch_mse
psnr = psnr + (10 * math.log10(1 / (batch_mse + 1.0e-6))
psnr = psnr + (10 * math.log10(1 / (batch_mse + 1.0e-6)))
loss_count = loss_count + 1
if loss_count % 10 == 0 then
xlua.progress(t, #data)