Chore: Change lr scheduler

Flippchen · Feb 8, 2024 · b52c613 · b52c613
1 parent df05e45
commit b52c613
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/train.py b/training/train.py
@@ -87,7 +87,7 @@
 
 # Define learning rate scheduler
 initial_learning_rate = 0.001
-lr_decay_steps = 1000
+lr_decay_steps = 10
 lr_decay_rate = 0.96
 lr_scheduler = tf.keras.optimizers.schedules.ExponentialDecay(
     initial_learning_rate,