diff --git a/dataset/dataset.py b/dataset/dataset.py
index 3d4c750c391d75849e1d73fe7f6ac97f429e8063..0ffebbb4dec481e5d0e04f7bc7f27939e2f4cd15 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 2e4b7fb8d3ec7fc5fe5e7cc3b95da1e2fd35be71..42a89ef28c27bd07d923f661c6c1fb44ab770977 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)