From 38431fb4c1cde164b3a62ea9ff12087160bbc0dc Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Wed, 12 Nov 2025 11:01:21 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A0=A1=E5=87=86=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/REPST/PEMS-BAY_paper.yaml | 60 ++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100755 config/REPST/PEMS-BAY_paper.yaml diff --git a/config/REPST/PEMS-BAY_paper.yaml b/config/REPST/PEMS-BAY_paper.yaml new file mode 100755 index 0000000..a0db8c9 --- /dev/null +++ b/config/REPST/PEMS-BAY_paper.yaml @@ -0,0 +1,60 @@ +basic: + dataset: "PEMS-BAY" + mode : "train" + device : "cuda:0" + model: "REPST" + +data: + add_day_in_week: true + add_time_in_day: true + column_wise: false + days_per_week: 7 + default_graph: true + horizon: 24 + lag: 24 + normalizer: std + num_nodes: 325 + steps_per_day: 288 + test_ratio: 0.2 + tod: false + val_ratio: 0.2 + sample: 1 + input_dim: 1 + batch_size: 16 + +model: + pred_len: 24 + seq_len: 24 + patch_len: 6 + stride: 7 + dropout: 0.2 + gpt_layers: 9 + d_ff: 128 + gpt_path: ./GPT-2 + d_model: 64 + n_heads: 1 + input_dim: 1 + t_max: 5 + +train: + batch_size: 16 + early_stop: true + early_stop_patience: 15 + epochs: 100 + grad_norm: false + loss_func: mse + lr_decay: true + lr_decay_rate: 0.75 + lr_decay_step: "5,20,40,70" + lr_init: 0.002 + max_grad_norm: 5 + real_value: true + seed: 2023 + weight_decay: 0 + debug: false + output_dim: 1 + log_step: 100 + plot: false + mae_thresh: None + mape_thresh: 0.001 +