From 6e19cd9c61971fbe1bddf6acaf994448217f323d Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 15:42:58 +0200
Subject: [PATCH] fix : dataloader batched
---
image_ref/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index acb26e5..8b6b095 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -20,7 +20,7 @@ def train_duo(model, data_train, optimizer, loss_function, epoch):
for imaer,imana, img_ref, label in data_train:
label = label.long()
- print(imaer.size)
+ print(imaer.shape)
if torch.cuda.is_available():
imaer = imaer.cuda()
imana = imana.cuda()
@@ -51,7 +51,7 @@ def test_duo(model, data_test, loss_function, epoch):
imana.transpose(0,1)
img_ref.transpose(0,1)
label.transpose(0,1)
- print(imaer.size)
+ print(imaer.shape)
label = label.long()
if torch.cuda.is_available():
imaer = imaer.cuda()
--
GitLab