From b7bf506ae572ec4247d6ef5696c39803409d162e Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Fri, 27 Sep 2024 17:56:02 +0200
Subject: [PATCH] revert tp training with 1 AR step by default

---
 neural_lam/train_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/neural_lam/train_model.py b/neural_lam/train_model.py
index 0913319e..8bad0f0c 100644
--- a/neural_lam/train_model.py
+++ b/neural_lam/train_model.py
@@ -125,9 +125,9 @@ def main(input_args=None):
     parser.add_argument(
         "--ar_steps_train",
         type=int,
-        default=3,
+        default=1,
         help="Number of steps to unroll prediction for in loss function "
-        "(default: 3)",
+        "(default: 1)",
     )
     parser.add_argument(
         "--control_only",