From 9ead61d792ab7c15aa0c874cb1a0cb7f96d13cb6 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 23 Jun 2025 10:53:06 +0200
Subject: [PATCH] float

---
 dataset/dataset.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/dataset/dataset.py b/dataset/dataset.py
index b78db35..9e91be8 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -236,6 +236,8 @@ class ImageFolderDuo(data.Dataset):
         impathAER, impathANA, target = self.imlist[index]
         imgAER = self.loader(impathAER)
         imgANA = self.loader(impathANA)
+        imgANA = imgANA.to(torch.float32)
+        imgAER = imgAER.to(torch.float32)
         if self.transform is not None:
             imgAER = self.transform(imgAER)
             imgANA = self.transform(imgANA)
-- 
GitLab