fixes #6641 (#6646)

deeplearning4j · Oct 30, 2018 · 08008bf · 08008bf
1 parent 5a41340
commit 08008bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...port/src/main/java/org/deeplearning4j/nn/modelimport/keras/utils/KerasOptimizerUtils.java b/...port/src/main/java/org/deeplearning4j/nn/modelimport/keras/utils/KerasOptimizerUtils.java
@@ -105,12 +105,13 @@ public static IUpdater mapOptimizer(Map<String, Object> optimizerConfig)
                 double beta1 = (double) optimizerParameters.get("beta_1");
                 double beta2 = (double) optimizerParameters.get("beta_2");
                 double epsilon = (double) optimizerParameters.get("epsilon");
-                double decay = (double) optimizerParameters.get("decay");
+                double scheduleDecay = (double) optimizerParameters.get("schedule_decay");
 
                 dl4jOptimizer = new Nadam.Builder()
                         .beta1(beta1).beta2(beta2)
                         .epsilon(epsilon).learningRate(lr)
-                        .learningRateSchedule(new InverseSchedule(ScheduleType.ITERATION, 1, decay, 1))
+                        .learningRateSchedule(new InverseSchedule(ScheduleType.ITERATION, 1,
+                                scheduleDecay, 1))
                         .build();
                 break;
             }