From ee1d3c654e4278c676bbb31e67769c3abe2c1005 Mon Sep 17 00:00:00 2001 From: Ldpe2G Date: Fri, 8 Apr 2016 23:06:15 +0800 Subject: [PATCH] fix the wrong requirement of FactorScheduler in LRScheduler.scala (#1807) fix the wrong requirement of FactorScheduler in LRScheduler.scala --- .../core/src/main/scala/ml/dmlc/mxnet/LRScheduler.scala | 4 ++-- 1 file changed, 2 insertions(+), 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 index 8cfee16aae80..192a4fd10cdc 100644 --- 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 = {