torch>=1.13.1
torchvision>=0.14.1
tqdm
wandb
