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

Fix division by zero error in validate()

Browse source
This commit is contained in:
nagadomi 2017-02-23 08:08:18 +09:00
parent d8b7df4505
commit a121eb39cb
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))
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)