From 109cff14988498c06e1519d4e1213f396f03267a Mon Sep 17 00:00:00 2001
From: beta <beta.underalpha@gmail.com>
Date: Wed, 25 May 2022 00:34:24 +0900
Subject: [PATCH] (#3) Defense: Apply ReduceLROnPlateau

---
 src/defense/models.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/defense/models.py b/src/defense/models.py
index 9a85081..e26129d 100644
--- a/src/defense/models.py
+++ b/src/defense/models.py
@@ -83,7 +83,11 @@ def predict(epoch, logs):
                     step=epoch,
                 )
 
-        return [keras.callbacks.LambdaCallback(on_epoch_end=predict)]
+        reduce_lr = keras.callbacks.ReduceLROnPlateau(
+            monitor="val_loss", factor=0.2, patience=5, min_lr=0.001
+        )
+
+        return [reduce_lr, keras.callbacks.LambdaCallback(on_epoch_end=predict)]
 
 
 class Denoiser(Defense):