Skip to content
Snippets Groups Projects
Commit f5fd0499 authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : dataloader batched

parent 21692d75
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
image_ref/dataset_ref.py
+ 2
− 2
View file @ f5fd0499
...@@ -225,8 +225,8 @@ class ImageFolderDuo_Batched(data.Dataset): ...@@ -225,8 +225,8 @@ class ImageFolderDuo_Batched(data.Dataset):
batched_im_ref = torch.concat(img_refs,dim=0) batched_im_ref = torch.concat(img_refs,dim=0)
batched_label = torch.tensor(label_refs) batched_label = torch.tensor(label_refs)
batched_imgAER = imgAER.repeat(len(self.classes)) batched_imgAER = imgAER.repeat(len(self.classes),1,1)
batched_imgANA = imgANA.repeat(len(self.classes)) batched_imgANA = imgANA.repeat(len(self.classes),1,1)
return batched_imgAER, batched_imgANA, batched_im_ref, batched_label return batched_imgAER, batched_imgANA, batched_im_ref, batched_label
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment