From d5e36476bf65025c47b5ad1aba6f7ca0f5709b5f Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Thu, 19 Jun 2025 08:52:55 +0200
Subject: [PATCH] fix : all tensor to float32

---
 dataset/dataset.py | 2 ++
 main.py            | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/dataset/dataset.py b/dataset/dataset.py
index 0ffebbb..32e96f4 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -260,6 +260,8 @@ class ImageFolderDuoPatch(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)
         shape_ana = imgANA.shape
         shape_aer = imgAER.shape
         if self.transform is not None:
diff --git a/main.py b/main.py
index 42a89ef..b4832c5 100644
--- a/main.py
+++ b/main.py
@@ -209,7 +209,7 @@ def run_duo(args):
     else :
         model = Classification_model_duo_pretrained(model = args.model, n_class=len(data_train.dataset.classes))
 
-    model = model.double()
+    model = model.to(torch.float32)
     #load weight
     if args.pretrain_path is not None :
         load_model(model,args.pretrain_path)
-- 
GitLab