Ever wanted to use Vision Transformers with TFHub and Keras? Well, pull your socks up now and get started. 16 different models are available for classification and fine-tuning. More details:

  
      

      github.com
  

  
    

GitHub - sayakpaul/ViT-jax2tf: This repository hosts code for converting the... 71


  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.  - GitHub - sayakpaul/ViT-jax2tf: This repository hosts code for converting the original Vision...

  

  
    
    
  

  


The story does not end here. Using this notebook you can convert any supported model from the AugReg pool (> 50,000 models!) and use that inside TFHub and Keras: https://colab.research.google.com/github/sayakpaul/ViT-jax2tf/blob/main/conversion.ipynb 21.

carbon1408×616 75.1 KB