diff --git a/.vscode/launch.json b/.vscode/launch.json index f8f45f8..e92ed4c 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -14,6 +14,14 @@ "console": "integratedTerminal", "args": "--config ./config/STID/PEMS-BAY.yaml" }, + { + "name": "STID: METR-LA", + "type": "debugpy", + "request": "launch", + "program": "run.py", + "console": "integratedTerminal", + "args": "--config ./config/STID/METR-LA.yaml" + }, { "name": "STID: PEMSD4", "type": "debugpy", diff --git a/config/STID/METR-LA.yaml b/config/STID/METR-LA.yaml new file mode 100755 index 0000000..5c43f0d --- /dev/null +++ b/config/STID/METR-LA.yaml @@ -0,0 +1,67 @@ +basic: + dataset: "METR-LA" + mode: "train" + device: "cuda:0" + model: "STID" + seed: 2023 + +data: + num_nodes: 207 + lag: 24 + horizon: 24 + val_ratio: 0.2 + test_ratio: 0.2 + tod: False + normalizer: std + column_wise: False + default_graph: True + add_time_in_day: True + add_day_in_week: True + steps_per_day: 288 + days_per_week: 7 + input_dim: 1 + output_dim: 1 + batch_size: 64 + +model: + input_dim: 3 + output_dim: 1 + history: 24 + horizon: 24 + num_nodes: 207 + input_len: 24 + embed_dim: 32 + output_len: 24 + num_layer: 3 + if_node: True + node_dim: 32 + if_T_i_D: True + if_D_i_W: True + temp_dim_tid: 32 + temp_dim_diw: 32 + time_of_day_size: 288 + day_of_week_size: 7 + batch_size: 64 + + +train: + loss_func: mae + seed: 1 + batch_size: 64 + epochs: 300 + lr_init: 0.002 + weight_decay: 0.0001 + lr_decay: False + lr_decay_rate: 0.3 + lr_decay_step: "1,50,80" + early_stop: True + early_stop_patience: 15 + grad_norm: False + max_grad_norm: 5 + real_value: True + debug: true + output_dim: 1 + mae_thresh: null + mape_thresh: 0.0 + log_step: 200 + plot: False