From e5217cab8595dbbe712d2a0c89f32aabd4091265 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 14 Apr 2025 10:27:57 +0200
Subject: [PATCH] add : model.float()
---
image_ref/main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index 8e22bdc..6cf959d 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')
--
GitLab