Hi Kevin,
In  your ‘phase_dataset’, you should have :   datasets.ImageFolder(phase_dir, transform=phase_transforms)
Check out: https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html#load-data 14