From b4cb160f6a913764e30f7e5fdd42373220f50fd9 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 15:49:59 +0200
Subject: [PATCH] fix : dataloader batched
---
image_ref/main.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index 003a807..062e538 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -47,10 +47,11 @@ def test_duo(model, data_test, loss_function, epoch):
param.requires_grad = False
for imaer,imana, img_ref, label in data_test:
- imaer=imaer.transpose(0,1)
- imana=imana.transpose(0,1)
- img_ref=img_ref.transpose(0,1)
- label=label.transpose(0,1)
+ imaer = imaer.transpose(0,1)
+ imana = imana.transpose(0,1)
+ img_ref = img_ref.transpose(0,1)
+ label = label.transpose(0,1)
+ label = label.squeeze()
print(label.shape)
label = label.long()
if torch.cuda.is_available():
--
GitLab