From ca3d148618bf37946f10b8e033955f1d9c17b07b Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 18 Jun 2025 17:00:18 +0200
Subject: [PATCH] fix : float model double data
---
dataset/dataset.py | 4 ++--
main.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/dataset/dataset.py b/dataset/dataset.py
index 3d4c750..0ffebbb 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -258,8 +258,8 @@ class ImageFolderDuoPatch(data.Dataset):
def __getitem__(self, index):
impathAER, impathANA, target = self.imlist[index]
- imgAER = self.loader(impathAER).float()
- imgANA = self.loader(impathANA).float()
+ imgAER = self.loader(impathAER)
+ imgANA = self.loader(impathANA)
shape_ana = imgANA.shape
shape_aer = imgAER.shape
if self.transform is not None:
diff --git a/main.py b/main.py
index 2e4b7fb..42a89ef 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.to(torch.float)
+ model = model.double()
#load weight
if args.pretrain_path is not None :
load_model(model,args.pretrain_path)
--
GitLab