czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix DSANET

This commit is contained in:
czzhangheng 2025-03-03 15:41:14 +08:00
parent 18aaa32c10
commit f35bfbf75d
3 changed files with 37 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
config/DSANET/PEMSD3.yaml
View File

@ -15,11 +15,20 @@ data:
model: model:
batch_size: 64 batch_size: 64
window: 12
local: 3
n_multiv: 358
n_kernels: 32
w_kernel: 1
d_model: 512
d_inner: 2048
n_layers: 6
n_head: 8
d_k: 64
d_v: 64
drop_prob: 0.1
input_dim: 1 input_dim: 1
output_dim: 1 output_dim: 1
num_channels: "16,32,1"
kernel_size: 2
dropout: 0.2
train: train:
loss_func: mae loss_func: mae

17
config/DSANET/PEMSD7.yaml
View File

@ -14,12 +14,21 @@ data:
days_per_week: 7 days_per_week: 7
model: model:
batch_size: 64 batch_size: 16
window: 12
local: 3
n_multiv: 883
n_kernels: 32
w_kernel: 1
d_model: 128
d_inner: 256
n_layers: 3
n_head: 4
d_k: 16
d_v: 16
drop_prob: 0.1
input_dim: 1 input_dim: 1
output_dim: 1 output_dim: 1
num_channels: "16,32,1"
kernel_size: 2
dropout: 0.2
train: train:
loss_func: mae loss_func: mae

15
config/DSANET/PEMSD8.yaml
View File

@ -15,11 +15,20 @@ data:
model: model:
batch_size: 64 batch_size: 64
window: 12
local: 3
n_multiv: 170
n_kernels: 32
w_kernel: 1
d_model: 512
d_inner: 2048
n_layers: 6
n_head: 8
d_k: 64
d_v: 64
drop_prob: 0.1
input_dim: 1 input_dim: 1
output_dim: 1 output_dim: 1
num_channels: "16,32,1"
kernel_size: 2
dropout: 0.2
train: train:
loss_func: mae loss_func: mae