add : model.float()

image_ref/main.py

@@ -107,7 +107,7 @@ def run_duo(args):
 
     # load model
     model = Classification_model_duo_contrastive(model=args.model, n_class=2)
-    model.double()
+    model.float()
     # load weight
     if args.pretrain_path is not None:
         print('Model weight loaded')