@InnovArul Thanks,
Then, how can I make a dataloader using this dataset? I am using this code
# 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)

But getting this error
AttributeError: 'CustomDataSet' object has no attribute 'size'
Could you tell me, how can I solve this problem?