numpy==1.21.2
torch==1.9.0
e2cnn==0.1.9
torchvision==0.10.0
matplotlib==3.4.3
scikit-learn==0.24.2
scipy==1.7.1
seaborn==0.11.2
