torch
torchvision
wandb
tqdm
