I am training a Variational Autoencoder. I have a dataset of RGB images with resolution 256*256
images = torchvision.datasets.ImageFolder(root = 'data/', transform=transforms.Compose([transforms.ToTensor()]))
data_loader = torch.utils.data.DataLoader(images, batch_size=32, shuffle=True)

However, my batches are turned into a list of tensors. As a result I cannot pass it to my model for training.
What kind of adjustment should I do in order for my model to be able to train? Thanks in advance