nagadomi пре 8 година
родитељ
комит
39c7ce1ea7
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      train.lua

+ 1 - 1
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)