From 21692d759d074eb20c5e67f72a8e55f389251c5b Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 15:17:01 +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 738820a..ae54025 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.Tensor.repeat(len(self.classes))
- batched_imgANA = imgANA.Tensor.repeat(len(self.classes))
+ batched_imgAER = imgAER.repeat(len(self.classes))
+ batched_imgANA = imgANA.repeat(len(self.classes))
return batched_imgAER, batched_imgANA, batched_im_ref, batched_label
--
GitLab