Yes I was actually! Downgrading to torchvision 0.8.2 actually fixed the issue.
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html 
Think there is something in torchvision 0.9.0 that doesn’t play nice with wandb.watch? Just searched and found this issue on wandb 24 too incase you hadn’t seen it