original

train.py

@@ -75,7 +75,7 @@ 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.AdamW(model.parameters(), lr=lr, weight_decay = 0.1)
+    optimizer = optim.AdamW(model.parameters(), lr=lr, weight_decay = 0.05)
     # optimizer = optim.SGD(model.parameters(), lr=lr)
 
     # --- training ---