torch~=1.5.0
torchvision~=0.6.0
scikit-learn>=0.21.1
tqdm>=4.24.0
visdom>=0.1.8.9
numpy>=1.16.4
psutil>=5.4.7
scipy>=1.3.1
