diff --git a/config/STAEFormer/METR-LA.yaml b/config/STAEFormer/METR-LA.yaml index 003e50e..6fa708e 100644 --- a/config/STAEFormer/METR-LA.yaml +++ b/config/STAEFormer/METR-LA.yaml @@ -9,9 +9,9 @@ data: batch_size: 16 column_wise: false days_per_week: 7 - horizon: 12 + horizon: 24 input_dim: 1 - lag: 12 + lag: 24 normalizer: std num_nodes: 207 steps_per_day: 288 @@ -23,13 +23,13 @@ model: dow_embedding_dim: 24 dropout: 0.1 feed_forward_dim: 256 - in_steps: 12 + in_steps: 24 input_dim: 1 input_embedding_dim: 24 num_heads: 4 num_layers: 3 num_nodes: 207 - out_steps: 12 + out_steps: 24 output_dim: 1 spatial_embedding_dim: 0 steps_per_day: 288