I trained a model with Colab TPU using torch XLA, every time I load the saved weight I got this error.
My pytorch version is 1.5.0.
I can’t even install the XLA in my machine. And can’t find any way to install the 1.6.
How to solve this and what’s the reason behind this feature?