Where do I find the new ViT baseline param?

[sayakpaul]

Regarding https://github.com/google-research/big_vision#vit-baseline

[xiaohuazhai]

It's available here:

https://github.com/google-research/big_vision/blob/main/big_vision/configs/vit_s16_i1k.py

You can always find all our configs under configs folder.

[sayakpaul]

Sorry, I should have been clearer. By param, I meant the pre-trained params. Are they available yet?

[akolesnikoff]

It is an oversight from our side: we will publish it soon. Leaving the issue open until then.

[sayakpaul]

Thanks!

As soon as they're available I will try to port them to a TensorFlow / Keras friendly model for wider community adoption.

[akolesnikoff]

Models have been landed, see

big_vision/big_vision/models/vit.py

Line 328 in 6c376d6

"i1k-s16-90ep": "gs://big_vision/vit_s16_i1k_90ep.npz",

Closing the issue, feel free to reopen if you have any questions.