torchvision>=0.11
numpy>=1.21
PyYAML
scipy
torch>=1.9
