argparse
torch==1.8.0
h5py==3.2.0
numpy==1.20.1
scikit-learn==0.24.1
scipy==1.6.2
