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