diff --git a/Code/python/src/aiRNN/dataloader.py b/Code/python/src/aiRNN/dataloader.py index 2a6ed94..a88da3b 100644 --- a/Code/python/src/aiRNN/dataloader.py +++ b/Code/python/src/aiRNN/dataloader.py @@ -143,11 +143,8 @@ 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]