torch>=1.5.0
matplotlib
numpy
tqdm
scipy
pyyaml
h5py
