accelerate
torch>=1.13
torchvision
tqdm
