Parameters for torch.utils.data.TensorDataset

1

training_set = TensorDataset(training_img_set,training_label_set)

I wanted to use this line of code to wrap my image data and labels into one tensor, but the variable training_img_set is in the form of a list of images which gives an error. How do I convert it to a tensor format so I can use it as a parameter for the function?

2

@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