torch>=1.7.1
torchvision>=0.8.2
numpy
pandas
scikit-learn
pynvml
tqdm
scikit-learn
munch
