torch
torchvision
tqdm
wandb
matplotlib
numpy
perforatedbp