torch==1.8.1+cu101
torchvision==0.9.1+cu101
torchaudio==0.8.1
wandb
tqdm
numpy
matplotlib
