torch>=1.7.0
torchvision>=0.8.1
tensorboard>=1.15.0
scikit-learn>=0.20.3
