scikit-learn==0.22.1
torch==1.11.0
torchvision==0.12.0
einops==0.3.2
pickle5==0.0.11
numpy==1.21.2
fitlog
requests
matplotlib
seaborn