KerasHub 模型架构
以下是 KerasHub 支持的模型架构列表。这些模型可以通过两种方式创建
- 通过
from_preset()构造函数，它实例化一个具有预训练配置、词汇表和权重的对象。 - 通过用户控制的自定义配置。
有关 Keras 团队直接提供的可用预训练模型预设的完整列表，请参阅预训练模型页面。
Albert
- AlbertTokenizer
- AlbertBackbone 模型
- AlbertTextClassifier 模型
- AlbertTextClassifierPreprocessor 层
- AlbertMaskedLM 模型
- AlbertMaskedLMPreprocessor 层
Bart
Bert
- BertTokenizer
- BertBackbone 模型
- BertTextClassifier 模型
- BertTextClassifierPreprocessor 层
- BertMaskedLM 模型
- BertMaskedLMPreprocessor 层
Bloom
DebertaV3
- DebertaV3Tokenizer
- DebertaV3Backbone 模型
- DebertaV3TextClassifier 模型
- DebertaV3TextClassifierPreprocessor 层
- DebertaV3MaskedLM 模型
- DebertaV3MaskedLMPreprocessor 层
DeepLabV3
- DeepLabV3ImageConverter
- DeepLabV3Backbone 模型
- DeepLabV3ImageSegmenter 模型
- DeepLabV3ImageSegmenterPreprocessor 层
DenseNet
- DenseNetImageConverter
- DensNetBackbone 模型
- DenseNetImageClassifier 模型
- DenseNetImageClassifierPreprocessor 层
DistilBert
- DistilBertTokenizer
- DistilBertBackbone 模型
- DistilBertTextClassifier 模型
- DistilBertTextClassifierPreprocessor 层
- DistilBertMaskedLM 模型
- DistilBertMaskedLMPreprocessor 层
Electra
Falcon
FNet
- FNetTokenizer
- FNetBackbone 模型
- FNetTextClassifier 模型
- FNetTextClassifierPreprocessor 层
- FNetMaskedLM 模型
- FNetMaskedLMPreprocessor 层
Gemma
GPT2
Llama
Llama3
Mistral
MiT
OPT
PaliGemma
Phi3
ResNet
Roberta
- RobertaTokenizer
- RobertaBackbone 模型
- RobertaTextClassifier 模型
- RobertaTextClassifierPreprocessor 层
- RobertaMaskedLM 模型
- RobertaMaskedLMPreprocessor 层
Segment Anything Model
- SAMImageConverter
- SAMBackbone 模型
- SAMImageSegmenter 模型
- SAMImageSegmenterPreprocessor 层
- SAMPromptEncoder 层
- SAMMaskDecoder 层
Stable Diffusion 3
- SAMImageConverter
- StableDiffusion3Backbone 模型
- StableDiffusion3TextToImage 模型
- StableDiffusion3TextToImagePreprocessor 层
- StableDiffusion3ImageToImage 模型
- StableDiffusion3Inpaint 模型