numpy
torch==1.8.0
torchvision==0.9.0
scipy
matplotlib
tqdm
pandas

