fix : float model double data

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:

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)