Click==7.0
colorama==0.4.3
matplotlib==3.5.1
numpy==1.17.4
setuptools==45.2.0
torch==1.8.1
torchvision==0.11.3
tqdm==4.62.3
wandb==0.12.11
