LiT training code

[ayyar]

Is there code to train or fine tune LiT models somwhere?

Notably, this colab just has code to use one of the existing models.
https://colab.research.google.com/github/google-research/vision_transformer/blob/main/lit.ipynb

[HuangChiEn]

Is there code to train or fine tune LiT models somwhere?

Notably, this colab just has code to use one of the existing models. https://colab.research.google.com/github/google-research/vision_transformer/blob/main/lit.ipynb

I'm also looking for the training code...orz, seems not really well-documented