From b43f982472efa306a5a5f0e3e09b5fd3b9076cee Mon Sep 17 00:00:00 2001 From: sanaAyrml Date: Mon, 6 Jan 2025 23:33:26 +0100 Subject: [PATCH] update rxrx1 dataset --- research/rxrx1/data/dataset.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/research/rxrx1/data/dataset.py b/research/rxrx1/data/dataset.py index ca3f73baa..0d17521ad 100644 --- a/research/rxrx1/data/dataset.py +++ b/research/rxrx1/data/dataset.py @@ -44,8 +44,7 @@ def __getitem__(self, idx: int) -> tuple[torch.Tensor, int, int]: images.append(image) concatenated_image = torch.cat(images, dim=0) - print(concatenated_image.shape) - return concatenated_image, label, row["sirna_id"] + return concatenated_image, label def load_image(self, path: str) -> torch.Tensor: if not Path(path).exists():