From e6cc05685830badf47162ca0af571070cdb2c6b3 Mon Sep 17 00:00:00 2001 From: Valeriu Lacatusu Date: Fri, 19 Oct 2018 13:41:13 +0200 Subject: [PATCH] Update ReduceLROnPlateau callback parameter name keras warning: `epsilon` argument is deprecated and will be removed, use `min_delta` instead. --- keras_retinanet/bin/train.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/keras_retinanet/bin/train.py b/keras_retinanet/bin/train.py index 7f12ba4aa..9e9a8443d 100755 --- 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