.

.gitignore

@@ -1 +1 @@
-__pycache__/
+__pycache__/

train.py

@@ -74,7 +74,8 @@ def train(symbol: str, seq_len: int = 60, epochs: int = 5, batch_size: int = 32,
     # --- model ---
     model = StockLSTM(input_dim=1, hidden_dim=64, num_layers=2, dropout=0.2).to(device)
     criterion = nn.MSELoss()
-    optimizer = optim.Adam(model.parameters(), lr=lr)
+    # optimizer = optim.Adam(model.parameters(), lr=lr)
+    optimizer = optim.SGD(model.parameters(), lr=lr)
 
     # --- training ---
     model.train()