torch>=1.2
torchvision
pandas
numpy
matplotlib
dill
tqdm>=4.38
natsort
scikit-learn
torchcontrib

[TBX]
tensorboardX
