Battle against a True Hero pt 2

This commit is contained in:
2026-01-24 20:26:38 -06:00
parent ae13d23a80
commit 35de11a12d

View File

@@ -206,4 +206,4 @@ if __name__ == "__main__":
print(f"\nModel initialized with {sum(p.numel() for p in model.parameters()):,} parameters") print(f"\nModel initialized with {sum(p.numel() for p in model.parameters()):,} parameters")
print("Starting training...\n") 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)