torch torchvision wandb perforatedai