argparse
numpy>=1.18.4
scikit-learn>=0.23.1
scipy>=1.4.1
torch>=1.5.0
pytorch-lightning>=1.0.3
