accelerate
torchvision
datasets