diff --git a/image_ref/main.py b/image_ref/main.py
index 8e22bdc73bd45aaab2c086962e2304c677667e01..6cf959da2921624f146436fcb8a3706f3c84374a 100644
--- a/image_ref/main.py
+++ b/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')