Found bug?!

2025-11-27 08:54:37 +01:00
parent 4be4bd79a2
commit c120deb9cc
1 changed files with 2 additions and 2 deletions
@@ -129,8 +129,8 @@ class SaveDataset(Dataset):
    def __init__(self, data_dict, step=1, start_offset=0, end_offset=0):
        super().__init__()
        self.step = step
-        self.start_offset = start_offset // step
-        self.end_offset = end_offset // step
+        self.start_offset = start_offset
+        self.end_offset = end_offset

        # Keep everything **on CPU**
        self.X_feat = data_dict["X_feat"]