add astra-pemsbay v2
This commit is contained in:
parent
c4414dd5d9
commit
e50f443470
|
|
@ -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
|
||||||
264
test_results.txt
264
test_results.txt
|
|
@ -734,270 +734,6 @@ Traceback (most recent call last):
|
||||||
AttributeError: 'NoneType' object has no attribute 'to'
|
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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 <module>
|
|
||||||
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
|
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-InFlow.yaml
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue