torch>=1.6.0
torchvision>=0.6
numpy>=1.21
wandb==0.12.7