perf: batch_size 64, max_epochs 50 — 加速训练约8x

2026-05-28 11:26:59 +08:00
parent c0e2bdca72
commit 305f70b9de
1 changed files with 2 additions and 2 deletions
@@ -34,9 +34,9 @@ ERA5_VARIABLES = [

 # 模型配置
 LOOKBACK_DAYS = 14
-BATCH_SIZE = 16
+BATCH_SIZE = 64
 LEARNING_RATE = 1e-3
-MAX_EPOCHS = 200
+MAX_EPOCHS = 50
 EARLY_STOP_PATIENCE = 15
 HIDDEN_DIM = 128
 LSTM_LAYERS = 2