From e50f44347013574f8441cf4726811e8cb3863bf0 Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Tue, 2 Dec 2025 09:40:24 +0800 Subject: [PATCH] add astra-pemsbay v2 --- config/AEPSA/v2_PEMS-BAY.yaml | 54 +++++++ test_results.txt | 264 ---------------------------------- 2 files changed, 54 insertions(+), 264 deletions(-) create mode 100755 config/AEPSA/v2_PEMS-BAY.yaml diff --git a/config/AEPSA/v2_PEMS-BAY.yaml b/config/AEPSA/v2_PEMS-BAY.yaml new file mode 100755 index 0000000..7b5c97e --- /dev/null +++ b/config/AEPSA/v2_PEMS-BAY.yaml @@ -0,0 +1,54 @@ +basic: + dataset: PEMS-BAY + device: cuda:0 + mode: train + model: AEPSA_v2 + seed: 2023 + +data: + batch_size: 16 + column_wise: false + days_per_week: 7 + horizon: 24 + input_dim: 1 + lag: 24 + normalizer: std + num_nodes: 325 + steps_per_day: 288 + test_ratio: 0.2 + val_ratio: 0.2 + +model: + d_ff: 128 + d_model: 64 + dropout: 0.2 + gpt_layers: 9 + gpt_path: ./GPT-2 + input_dim: 1 + n_heads: 1 + num_nodes: 325 + patch_len: 6 + pred_len: 24 + seq_len: 24 + stride: 7 + word_num: 1000 + +train: + batch_size: 16 + debug: false + early_stop: true + early_stop_patience: 15 + epochs: 100 + grad_norm: false + log_step: 100 + loss_func: mae + lr_decay: true + lr_decay_rate: 0.3 + lr_decay_step: 5,20,40,70 + lr_init: 0.003 + mae_thresh: None + mape_thresh: 0.001 + max_grad_norm: 5 + output_dim: 1 + plot: false + weight_decay: 0 diff --git a/test_results.txt b/test_results.txt index 6b5c248..6116217 100644 --- a/test_results.txt +++ b/test_results.txt @@ -734,270 +734,6 @@ Traceback (most recent call last): AttributeError: 'NoneType' object has no attribute 'to' ``` -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-InFlow.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD4.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/METR-LA.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/AirQuality.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-OutFlow.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD8.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(L).yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD3.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/SolarEnergy.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/Hainan.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7.yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - -#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(M).yaml - -``` - -Traceback (most recent call last): - File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in - main() - File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main - model = init.init_model(args) - ^^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model - model = model_selector(args).to(device) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector - return TWDGCN(model_config) - ^^^^^^^^^^^^^^^^^^^^ - File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__ - self.num_node = args["num_nodes"] - ~~~~^^^^^^^^^^^^^ -KeyError: 'num_nodes' -``` - #### /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-InFlow.yaml ```