Update ReduceLROnPlateau callback parameter name

keras warning: `epsilon` argument is deprecated and will be removed, use `min_delta` instead.
huynhtnhut97 · Oct 19, 2018 · e6cc056 · e6cc056
1 parent 79c80bf
commit e6cc056
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/keras_retinanet/bin/train.py b/keras_retinanet/bin/train.py
@@ -193,14 +193,14 @@ def create_callbacks(model, training_model, prediction_model, validation_generat
         callbacks.append(checkpoint)
 
     callbacks.append(keras.callbacks.ReduceLROnPlateau(
-        monitor  = 'loss',
-        factor   = 0.1,
-        patience = 2,
-        verbose  = 1,
-        mode     = 'auto',
-        epsilon  = 0.0001,
-        cooldown = 0,
-        min_lr   = 0
+        monitor    = 'loss',
+        factor     = 0.1,
+        patience   = 2,
+        verbose    = 1,
+        mode       = 'auto',
+        min_delta  = 0.0001,
+        cooldown   = 0,
+        min_lr     = 0
     ))
 
     return callbacks