From cae27a9125a71d2bd82d95c935648ee78cdd8175 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 14 Apr 2025 10:53:06 +0200
Subject: [PATCH] add : out_dit for fig as args
---
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 a6b0988..cbb8986 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.float()
+ model.double()
# load weight
if args.pretrain_path is not None:
print('Model weight loaded')
--
GitLab