revert tp training with 1 AR step by default

mllam · Sep 27, 2024 · b7bf506 · b7bf506
1 parent ac7e46a
commit b7bf506
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",