fix clip bug in weight_clip.py

hillbig · Feb 14, 2016 · 14eef4c · 14eef4c
1 parent 935364b
commit 14eef4c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/weight_clip.py b/weight_clip.py
@@ -1,3 +1,4 @@
+import numpy
 from chainer import cuda
 
 class WeightClip(object):
@@ -34,6 +35,6 @@ def __call__(self, opt):
             p = param.data
             with cuda.get_device(p) as dev:
                 if int(dev) == -1:
-                    p = numpy.min(self.high, numpy.max(self.low, p))
+                    numpy.clip(p, self.low, self.high)
                 else:
                     kernel(self.low, self.high, p)