From ef6067676b6f54b8a78b4ecb68825d781eac1f3e Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 18 Apr 2025 11:41:59 +0200
Subject: [PATCH] fix : error device cuda
---
image_ref/main_ray.py | 3 ---
1 file changed, 3 deletions(-)
diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py
index 03d9016..d0e2360 100644
--- a/image_ref/main_ray.py
+++ b/image_ref/main_ray.py
@@ -83,9 +83,6 @@ def train_model(config,args):
imana = imana.cuda()
img_ref = img_ref.cuda()
label = label.cuda()
- if torch.cuda.device_count() > 1:
- pred_logits = model.module.forward(imaer, imana, img_ref)
- else:
pred_logits = model.forward(imaer, imana, img_ref)
--
GitLab