From a790e0c6360878ef2ed3426be9f8bf06bf04a4a9 Mon Sep 17 00:00:00 2001 From: Schneider Leo <leo.schneider@etu.ec-lyon.fr> Date: Tue, 1 Apr 2025 15:45:03 +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 8b6b0957..f6654fdd 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -47,10 +47,10 @@ def test_duo(model, data_test, loss_function, epoch): param.requires_grad = False for imaer,imana, img_ref, label in data_test: - imaer.transpose(0,1) - imana.transpose(0,1) - img_ref.transpose(0,1) - 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) print(imaer.shape) label = label.long() if torch.cuda.is_available(): -- GitLab