torch
torchvision
numpy
tensorboard
tqdm
h5py
scipy
scikit-image
pyyaml
