requirements.txt

# pip install -r requirements.txt

numpy>=1.18.5
torch>=1.7.0
torchvision>=0.8.1
scikit-learn==1.2.2