From 07d7d4385748fcad1faf1433cc1cb1094326780c Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Wed, 3 Dec 2025 18:51:54 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E9=85=8D=E7=BD=AE=EF=BC=8CME?= =?UTF-8?q?TR-LA=20-STAEFormer?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/STAEFormer/METR-LA.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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