torch==1.9.0
torchvision
tensorboard
scikit-learn
