Pillow
torch
torchmetrics
torchvision
tqdm
