h5py
pyyaml
matplotlib
torch