@InnovArul, I am using transform in my CustomDataSet. So, I am guessing that the code converting the dataset into tensor (I am not sure).



 akib62:

# Giving the path 
  train_data_tensor = CustomDataSet(train_data_dir, transform=transform)



train_label_price_tensor is tensor. I made it using
train_label_price_tensor = torch.tensor(label_price.values)
Is it possible to create a CustomDataSet into Tensor?