diff --git a/src/utils/config.py b/src/utils/config.py index 1b7e4f4..aa3c600 100644 --- a/src/utils/config.py +++ b/src/utils/config.py @@ -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