fix the wrong requirement of FactorScheduler in LRScheduler.scala (ap…

…ache#1807) fix the wrong requirement of FactorScheduler in LRScheduler.scala
liyangdev · Apr 8, 2016 · ee1d3c6 · ee1d3c6
1 parent 1a61815
commit ee1d3c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scala-package/core/src/main/scala/ml/dmlc/mxnet/LRScheduler.scala b/scala-package/core/src/main/scala/ml/dmlc/mxnet/LRScheduler.scala
@@ -38,8 +38,8 @@ class FactorScheduler(protected var step: Int, protected var factor: Float) exte
   protected var count: Int = 0
   private val logger = LoggerFactory.getLogger(classOf[FactorScheduler])
 
-  require(step < 1, "Schedule step must be greater or equal than 1 round")
-  require(factor >= 1.0, "Factor must be less than 1 to make lr reduce")
+  require(step >= 1, "Schedule step must be greater or equal than 1 round")
+  require(factor < 1.0, "Factor must be less than 1 to make lr reduce")
 
   def apply(numUpdate: Int): Float = {