@ksalkar where are you getting these list of images from? Can you paste the error here?
Have a look at this thread which might be of some help https://discuss.pytorch.org/t/make-a-tensordataset-and-dataloader-with-multiple-inputs-parameters/26605
@ksalkar where are you getting these list of images from? Can you paste the error here?
Have a look at this thread which might be of some help https://discuss.pytorch.org/t/make-a-tensordataset-and-dataloader-with-multiple-inputs-parameters/26605