WandB: https://wandb.ai/sayakpaul/shot-categorizer/runs/2ean2v4u