From 56260ac9dd343e78563e1d89c1d019564dc5a097 Mon Sep 17 00:00:00 2001 From: namsaraeva Date: Fri, 17 May 2024 10:05:09 +0200 Subject: [PATCH] rollback getitem --- src/sparcscore/ml/datasets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/sparcscore/ml/datasets.py b/src/sparcscore/ml/datasets.py index dde508b..2f936d9 100644 --- a/src/sparcscore/ml/datasets.py +++ b/src/sparcscore/ml/datasets.py @@ -318,10 +318,10 @@ def __getitem__(self, idx): if self.return_id: ids = int(data_item[3]) - sample = (t, torch.tensor(target), torch.tensor(ids)) # return data, target, and id + sample = (t, target, torch.tensor(ids)) # return data, target, and id elif self.return_fake_id: - sample = (t, torch.tensor(target), torch.tensor(0)) # return data, target, and fake id + sample = (t, target, torch.tensor(0)) # return data, target, and fake id else: - sample = (t, torch.tensor(target)) # return data and target + sample = (t, target) # return data and target return sample \ No newline at end of file