diff --git a/config/DSANET/PEMSD3.yaml b/config/DSANET/PEMSD3.yaml
index f274533..c87e1ef 100644
--- a/config/DSANET/PEMSD3.yaml
+++ b/config/DSANET/PEMSD3.yaml
@@ -15,11 +15,20 @@ data:
 
 model:
   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
   output_dim: 1
-  num_channels: "16,32,1"
-  kernel_size: 2
-  dropout: 0.2
 
 train:
   loss_func: mae
diff --git a/config/DSANET/PEMSD7.yaml b/config/DSANET/PEMSD7.yaml
index 124ee53..f10b5fe 100644
--- a/config/DSANET/PEMSD7.yaml
+++ b/config/DSANET/PEMSD7.yaml
@@ -14,12 +14,21 @@ data:
   days_per_week: 7
 
 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
   output_dim: 1
-  num_channels: "16,32,1"
-  kernel_size: 2
-  dropout: 0.2
 
 train:
   loss_func: mae
diff --git a/config/DSANET/PEMSD8.yaml b/config/DSANET/PEMSD8.yaml
index f752aba..4eb9219 100644
--- a/config/DSANET/PEMSD8.yaml
+++ b/config/DSANET/PEMSD8.yaml
@@ -15,11 +15,20 @@ data:
 
 model:
   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
   output_dim: 1
-  num_channels: "16,32,1"
-  kernel_size: 2
-  dropout: 0.2
 
 train:
   loss_func: mae