Using Huggingface for computer vision (Tensorflow)?


I would like to to use huggingface to train some computer vision models. The issue is that I use tensorflow and I can only see pytorch models on the hub.

Is tensorflow supported for computer vision? Is there a TF notebook I could use to train my own models?


We’re currently adding a TF implementation of the Vision Transformer: Add TFViTModel by ydshieh · Pull Request #13778 · huggingface/transformers · GitHub

This will also make it easier to add the other vision models (DeiT, BEiT), which are very similar to ViT.

1 Like

thanks @nielsr !! super useful. I imagine there is a similar tensorflow notebook for audio recognition? :pray: :pray: :pray: