torch==1.7.0+cu110
numpy==1.19.5
scikit-learn==0.23.2