From 8453c2b482533e90e8f75884fc4ada9a4fdb4a6e Mon Sep 17 00:00:00 2001 From: Leif Denby Date: Fri, 27 Sep 2024 15:09:44 +0200 Subject: [PATCH] Update neural_lam/models/ar_model.py Co-authored-by: Joel Oskarsson --- neural_lam/models/ar_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py index ec1649a4..c1ceff81 100644 --- a/neural_lam/models/ar_model.py +++ b/neural_lam/models/ar_model.py @@ -291,7 +291,7 @@ def validation_step(self, batch, batch_idx): val_log_dict = { f"val_loss_unroll{step}": time_step_loss[step - 1] for step in self.args.val_steps_to_log - if step < len(time_step_loss) + if step <= len(time_step_loss) } val_log_dict["val_mean_loss"] = mean_loss self.log_dict(