torch == 1.8.1
torchvision == 0.9.1
tensorflow >= 2.0
numpy
pandas
scikit-learn
click
