diff --git a/train.lua b/train.lua index b3d9ddf..6379aeb 100644 --- a/train.lua +++ b/train.lua @@ -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)