Open dlmacedo opened 3 years ago
https://colab.research.google.com/github/google-research/vision_transformer/blob/master/vit_jax.ipynb
Hi,
I would like to ask whether you can run the code on colab tpu or your own tpu. I find an error when I try to run the code and I would like to ask whether you can give me some advice.
Best, Yong
https://colab.research.google.com/github/google-research/vision_transformer/blob/master/vit_jax.ipynb