diff --git a/rewrite_training.py b/rewrite_training.py
index fe1be64..8a11a07 100644
--- a/rewrite_training.py
+++ b/rewrite_training.py
@@ -206,4 +206,4 @@ if __name__ == "__main__":
     print(f"\nModel initialized with {sum(p.numel() for p in model.parameters()):,} parameters")
     print("Starting training...\n")
 
-    train_model(model, train_loader, val_loader, epochs=20, lr=0.002)
\ No newline at end of file
+    train_model(model, train_loader, val_loader, epochs=20, lr=0.001)
\ No newline at end of file