KerasHub Model Architectures
The following is a list of model architectures supported by KerasHub. These models can be created in two ways:
- Through the
from_preset()constructor, which instantiates an object with a pre-trained configurations, vocabularies, and weights. - Through custom configuration controlled by the user.
For the full list of available pretrained model presets shipped directly by the Keras team, see the Pretrained Models page.
Albert
- AlbertTokenizer
- AlbertBackbone model
- AlbertTextClassifier model
- AlbertTextClassifierPreprocessor layer
- AlbertMaskedLM model
- AlbertMaskedLMPreprocessor layer
Bart
Bert
- BertTokenizer
- BertBackbone model
- BertTextClassifier model
- BertTextClassifierPreprocessor layer
- BertMaskedLM model
- BertMaskedLMPreprocessor layer
Bloom
DebertaV3
- DebertaV3Tokenizer
- DebertaV3Backbone model
- DebertaV3TextClassifier model
- DebertaV3TextClassifierPreprocessor layer
- DebertaV3MaskedLM model
- DebertaV3MaskedLMPreprocessor layer
DeepLabV3
- DeepLabV3ImageConverter
- DeepLabV3Backbone model
- DeepLabV3ImageSegmenter model
- DeepLabV3ImageSegmenterPreprocessor layer
DenseNet
- DenseNetImageConverter
- DensNetBackbone model
- DenseNetImageClassifier model
- DenseNetImageClassifierPreprocessor layer
DistilBert
- DistilBertTokenizer
- DistilBertBackbone model
- DistilBertTextClassifier model
- DistilBertTextClassifierPreprocessor layer
- DistilBertMaskedLM model
- DistilBertMaskedLMPreprocessor layer
Electra
Falcon
FNet
- FNetTokenizer
- FNetBackbone model
- FNetTextClassifier model
- FNetTextClassifierPreprocessor layer
- FNetMaskedLM model
- FNetMaskedLMPreprocessor layer
Gemma
GPT2
Llama
Llama3
Mistral
MiT
OPT
PaliGemma
- PaliGemmaTokenizer
- PaliGemmaBackbone model
- PaliGemmaCausalLM model
- PaliGemmaCausalLMPreprocessor layer
Phi3
ResNet
- ResNetImageConverter
- ResNetBackbone model
- ResNetImageClassifier model
- ResNetImageClassifierPreprocessor layer
Roberta
- RobertaTokenizer
- RobertaBackbone model
- RobertaTextClassifier model
- RobertaTextClassifierPreprocessor layer
- RobertaMaskedLM model
- RobertaMaskedLMPreprocessor layer
Segment Anything Model
- SAMImageConverter
- SAMBackbone model
- SAMImageSegmenter model
- SAMImageSegmenterPreprocessor layer
- SAMPromptEncoder layer
- SAMMaskDecoder layer
Stable Diffusion 3
- SAMImageConverter
- StableDiffusion3Backbone model
- StableDiffusion3TextToImage model
- StableDiffusion3TextToImagePreprocessor layer
- StableDiffusion3ImageToImage model
- StableDiffusion3Inpaint model