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