From 107fdd0465de3ae5e6e721b41b618aa208fd9a84 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 15:39:20 +0200
Subject: [PATCH] fix : dataloader batched
---
image_ref/main.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index b4b6e22..12ebb40 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -46,10 +46,10 @@ def test_duo(model, data_test, loss_function, epoch):
param.requires_grad = False
for imaer,imana, img_ref, label in data_test:
- imaer.permute(0,1)
- imana.permute(0,1)
- img_ref.permute(0,1)
- label.permute(0,1)
+ imaer.transpose(0,1)
+ imana.transpose(0,1)
+ img_ref.transpose(0,1)
+ label.transpose(0,1)
label = label.long()
if torch.cuda.is_available():
imaer = imaer.cuda()
--
GitLab