It is confusing to me at this point.
I see that you have own dataset class CustomDataset. I also see the use of the TensorDataset.
You can use any object of a Dataset class as input to Dataloader.



 akib62:

# Making tensor using both image and labels tensor
train_tensor = TensorDataset(train_data_tensor, train_label_price_tensor)



Here, train_data_tensor, train_label_price_tensor are tensors? If there are tensors, I am not sure why the error talks about CustomDataSet.