numpy
torch>=1.8.1
torchvision>=0.9.1
scipy
sklearn
matplotlib
h5py
tqdm
