Unverified Kaydet (Commit) e6cc0568 authored tarafından Valeriu Lacatusu's avatar Valeriu Lacatusu Kaydeden (comit) GitHub

Update ReduceLROnPlateau callback parameter name

keras warning: `epsilon` argument is deprecated and will be removed, use `min_delta` instead.
üst 79c80bf3
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment