Debugging of mode 3

2025-11-27 08:52:43 +01:00
parent 2cd518321f
commit 4be4bd79a2
2 changed files with 7 additions and 4 deletions
@@ -239,14 +239,14 @@ def run_mode_3():

    for warm in [300, 600, 900, 1800]:
        for pred in [150, 300, 600, 900, 1800]:
-            start = 30 * 60 - warm
-            end = 30 * 60 - pred
+            start_offset = 30 * 60 - warm
+            end_offset = 30 * 60 - pred

            base_ds = dataloader.SaveDataset(
                torch.load("long_dataset.pt"),
                step=step,
-                start_offset=start,
-                end_offset=end,
+                start_offset=start_offset,
+                end_offset=end_offset,
            )

            train_ds, val_ds = split_dataset(base_ds)