Battle against a True Hero pt 2

2026-01-24 20:26:38 -06:00
parent ae13d23a80
commit 35de11a12d
1 changed files with 1 additions and 1 deletions
--- 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)
+    train_model(model, train_loader, val_loader, epochs=20, lr=0.001)