diff --git a/.vscode/launch.json b/.vscode/launch.json index 12e148f..6193b75 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -225,6 +225,14 @@ "program": "run.py", "console": "integratedTerminal", "args": "--config ./config/AEPSA/v2_METR-LA.yaml" - } + }, + { + "name": "AEPSA_v2: SolarEnergy", + "type": "debugpy", + "request": "launch", + "program": "run.py", + "console": "integratedTerminal", + "args": "--config ./config/AEPSA/v2_SolarEnergy.yaml" + }, ] } \ No newline at end of file diff --git a/config/AEPSA/v2_SolarEnergy.yaml b/config/AEPSA/v2_SolarEnergy.yaml new file mode 100644 index 0000000..26f286f --- /dev/null +++ b/config/AEPSA/v2_SolarEnergy.yaml @@ -0,0 +1,59 @@ +basic: + dataset: "SolarEnergy" + mode : "train" + device : "cuda:1" + model: "AEPSA_v2" + seed: 2023 + +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: 137 + steps_per_day: 24 + test_ratio: 0.2 + tod: false + val_ratio: 0.2 + sample: 1 + input_dim: 1 + batch_size: 64 + +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 + word_num: 1000 + num_nodes: 137 + +train: + batch_size: 64 + early_stop: true + early_stop_patience: 15 + epochs: 100 + grad_norm: false + loss_func: mae + lr_decay: true + lr_decay_rate: 0.3 + lr_decay_step: "5,20,40,70" + lr_init: 0.003 + max_grad_norm: 5 + weight_decay: 0 + debug: false + output_dim: 1 + log_step: 100 + plot: false + mae_thresh: None + mape_thresh: 0.001