Debugging of mode 3

This commit is contained in:
2025-11-27 08:52:43 +01:00
parent 2cd518321f
commit 4be4bd79a2
2 changed files with 7 additions and 4 deletions
+3
View File
@@ -143,8 +143,11 @@ class SaveDataset(Dataset):
t0 = self.start_offset
t1 = max_len - self.end_offset
self.idx_feat = torch.arange(t0, self.X_feat.shape[1], self.step)
print(self.idx_feat)
self.idx_time = torch.arange(t0, t1, self.step)
print(self.idx_time)
self.idx_out = torch.arange(0, self.Y_out.shape[1] - self.end_offset, self.step)
print(self.idx_out)
def __len__(self):
return self.X_feat.shape[0]