From f5fd04991c1f8c10d76fe7c54eed76e3ece7023a Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 15:26:06 +0200
Subject: [PATCH] fix : dataloader batched
---
image_ref/dataset_ref.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index ae54025..f0f329c 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -225,8 +225,8 @@ class ImageFolderDuo_Batched(data.Dataset):
batched_im_ref = torch.concat(img_refs,dim=0)
batched_label = torch.tensor(label_refs)
- batched_imgAER = imgAER.repeat(len(self.classes))
- batched_imgANA = imgANA.repeat(len(self.classes))
+ batched_imgAER = imgAER.repeat(len(self.classes),1,1)
+ batched_imgANA = imgANA.repeat(len(self.classes),1,1)
return batched_imgAER, batched_imgANA, batched_im_ref, batched_label
--
GitLab