numpy>=1.21.0
torch>=1.8.0
torchstat
tqdm
scipy
torchvision
