torch>=1.8.0
numpy>=1.21.0
torchstat
scipy
tqdm
