From 3d35a0fcd8982e6432ed6e136318467b564172d2 Mon Sep 17 00:00:00 2001 From: Schneider Leo <leo.schneider@etu.ec-lyon.fr> Date: Tue, 1 Apr 2025 15:37:38 +0200 Subject: [PATCH] fix : dataloader batched --- image_ref/main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/image_ref/main.py b/image_ref/main.py index d1b00ae..b4b6e22 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -19,7 +19,6 @@ def train_duo(model, data_train, optimizer, loss_function, epoch): param.requires_grad = True for imaer,imana, img_ref, label in data_train: - print(imaer.size()) label = label.long() if torch.cuda.is_available(): imaer = imaer.cuda() @@ -47,7 +46,10 @@ def test_duo(model, data_test, loss_function, epoch): param.requires_grad = False for imaer,imana, img_ref, label in data_test: - print(imaer.size()) + imaer.permute(0,1) + imana.permute(0,1) + img_ref.permute(0,1) + label.permute(0,1) label = label.long() if torch.cuda.is_available(): imaer = imaer.cuda() -- GitLab