[FIX](config) Bad source for ExponentialDecay

wbenbihi · Aug 25, 2022 · 6c05364 · 6c05364
1 parent 421d578
commit 6c05364
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config/config.default.yaml b/config/config.default.yaml
@@ -116,7 +116,7 @@ train:
   epoch_size: 1000
   batch_size: 512
   learning_rate:
-    source: "keras.optimizers.schedules.ExponentialDecay"
+    source: "keras.optimizers.schedules.learning_rate_schedule.ExponentialDecay"
     params:
       initial_learning_rate: 0.00025
       decay_steps: 1000