From cf638aea484104efb57c92727a7424407ae503e8 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Wed, 3 Oct 2018 18:34:26 +0900 Subject: [PATCH] Add default option for ClippedMSECriterion --- lib/ClippedMSECriterion.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/ClippedMSECriterion.lua b/lib/ClippedMSECriterion.lua index 204d156..a50b371 100644 --- a/lib/ClippedMSECriterion.lua +++ b/lib/ClippedMSECriterion.lua @@ -2,8 +2,8 @@ local ClippedMSECriterion, parent = torch.class('w2nn.ClippedMSECriterion','nn.C function ClippedMSECriterion:__init(min, max) parent.__init(self) - self.min = min - self.max = max + self.min = min or 0 + self.max = max or 1 self.diff = torch.Tensor() self.diff_pow2 = torch.Tensor() end