From 29063c8f3227bbf5fc054983d485155eab9cc06c Mon Sep 17 00:00:00 2001 From: joeloskarsson Date: Thu, 5 Dec 2024 14:16:17 +0100 Subject: [PATCH] Adjust forcing dimensionalities after fix --- neural_lam/models/ar_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py index 9773c7b1..1edb7931 100644 --- a/neural_lam/models/ar_model.py +++ b/neural_lam/models/ar_model.py @@ -109,7 +109,7 @@ def __init__( 2 * self.grid_output_dim + grid_static_dim # Temporal Embedding counts as one additional forcing_feature - + 2*(num_forcing_vars) + + (num_forcing_vars + 1) * (num_past_forcing_steps + num_future_forcing_steps + 1) ) @@ -144,7 +144,7 @@ def __init__( self.boundary_dim = ( boundary_static_dim # Temporal Embedding counts as one additional forcing_feature - + 2*(num_boundary_forcing_vars) + + (num_boundary_forcing_vars + 1) * (num_past_boundary_steps + num_future_boundary_steps + 1) )