Bagg:

Also, this error only comes up on my machine and it doesn’t while running my script on Google Colab.


I guess your local setup might be using an older torchvision release, so you might want to update it.