numpy==1.20.1
scikit_learn==1.2.2
submodlib==1.1.5
torch==1.9.1
torchdatasets==0.2.0
torchvision==0.10.1
tqdm==4.59.0
wandb==0.12.21
