suggest to reduce ar_steps and forcing window with small dataset

mllam · Nov 12, 2024 · a8362ce · a8362ce
1 parent b33e863
commit a8362ce
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/neural_lam/weather_dataset.py b/neural_lam/weather_dataset.py
@@ -50,8 +50,11 @@ def __init__(
             raise ValueError(
                 "The provided datastore only provides "
                 f"{len(self.da_state.time)} total time steps, which is too few "
-                "to create a single sample for the WeatherDataset configuration used"
-                f" in the `{split}` split."
+                "to create a single sample for the WeatherDataset "
+                f"configuration used in the `{split}` split. You could try "
+                "either reducing the number of autoregressive steps "
+                "(`ar_steps`) and/or the forcing window size "
+                "(`include_past_forcing` and `include_future_forcing`)"
             )
 
         # Set up for standardization