From 305f70b9de25884e38f0ccbda8aeb1d1eb0ad7e5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E8=88=AA=E5=AE=87?= <3364451258@qq.com> Date: Thu, 28 May 2026 11:26:59 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20batch=5Fsize=2064,=20max=5Fepochs=2050?= =?UTF-8?q?=20=E2=80=94=20=E5=8A=A0=E9=80=9F=E8=AE=AD=E7=BB=83=E7=BA=A68x?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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