@ptrblck, here I am doing transform. Does it mean, images are converted into tensor, not the CustomDataSet



 akib62:

#loading images using CustomDataSet
    train_data_tensor = CustomDataSet(train_data_dir, transform=transform)