From a121eb39cbe8de9d72a355c645e4e0c2ea5d280c Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Thu, 23 Feb 2017 08:08:18 +0900
Subject: [PATCH] Fix division by zero error in validate()

---
 train.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.lua b/train.lua
index 3f2eecb..b3d9ddf 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))
+      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)