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

fix : dataloader batched

parent 74e27806
No related branches found
No related tags found
No related merge requests found
......@@ -225,8 +225,8 @@ class ImageFolderDuo_Batched(data.Dataset):
batched_im_ref = torch.concat(img_refs,dim=0)
batched_label = torch.tensor(label_refs)
batched_imgAER = imgAER.Tensor.repeat(len(self.classes))
batched_imgANA = imgANA.Tensor.repeat(len(self.classes))
batched_imgAER = imgAER.repeat(len(self.classes))
batched_imgANA = imgANA.repeat(len(self.classes))
return batched_imgAER, batched_imgANA, batched_im_ref, batched_label
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment