فهرست منبع

Add default option for ClippedMSECriterion

nagadomi 6 سال پیش
والد
کامیت
cf638aea48
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  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