perf: 优化训练配置 — batch_size 16, alpha 0.5, max_epochs 200, 日志文件输出

2026-05-28 09:51:03 +08:00
parent fff2a7cabd
commit 0d178b0d57
2 changed files with 13 additions and 4 deletions
@@ -34,9 +34,9 @@ ERA5_VARIABLES = [

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