some update

intel-analytics · Jul 13, 2020 · 03a3a97 · 03a3a97
1 parent 9df3124
commit 03a3a97
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/zoo/src/main/scala/com/intel/analytics/zoo/pipeline/api/keras/optimizers/Adam.scala b/zoo/src/main/scala/com/intel/analytics/zoo/pipeline/api/keras/optimizers/Adam.scala
@@ -66,6 +66,7 @@ class Adam[@specialized(Float, Double) T: ClassTag](
     val beta1 = this.beta_1
     val beta2 = this.beta_2
     val eps = this.epsilon
+    val wd = this.wDecay
 
     val (fx, dfdx) = feval(parameter)
     val state = SGDRef.getstate(this)
@@ -81,8 +82,8 @@ class Adam[@specialized(Float, Double) T: ClassTag](
 
     val clr = - this.schedule.currentRate
 
-    if(wDecay > 0) {
-      dfdx.add(parameter * (ev.fromType(wDecay)))
+    if(wd > 0) {
+      dfdx.add(parameter * (ev.fromType(wd)))
     }
 
     /**