@pchandrasekaran thank you,
Yes, I was just simply trying to find the length of the dataset. Because I am using ToTensor at the time of transform. It means, my dataset is converting into Tensor formate, right? If ToTensor worked then, I should get the size, right?
However, I need to add data_tensor and label_tensor both at the time of dataloader. Code is given below:
# Making tensor using both image and labels tensor
train_tensor = TensorDataset(train_data_tensor, train_label_price_tensor)
# Making Train Dataloader
train_loader = DataLoader(train_tensor, batch_size= 1,
num_workers= 2, shuffle= True)
I am getting this error
AttributeError: 'CustomDataSet' object has no attribute 'size'
Could you tell me, how can I solve this problem?