Browse Source

Add default option for ClippedMSECriterion

nagadomi 6 years ago
parent
commit
cf638aea48
1 changed files with 2 additions and 2 deletions
  1. 2 2
      lib/ClippedMSECriterion.lua

+ 2 - 2
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