在 transformers 中实现了大量的算法模型类,有 Bert 模型的 BertModel 类,有 BART 模型的 BartModel 类,有 GPT 模型的 GPT2Model 类等。
AutoTokenizer 会根据 model 的类型自动选择合适的分词器。需要注意的是,预训练模型与分词器是配套使用的。如果使用了 cardiffnlp/twitter-roberta-base-sentiment-latest 模型,就应该使用 cardiffnlp/twitter-roberta-base-sentiment-latest 分词器,否则效果会很差。
| |
本篇主要是介绍了 transformers 中的 pipeline、模型类、分词器、模型配置类等。pipeline 提供了便捷的方式,将分词器、模型处理、后处理器等组合在一起,方便用户使用。AutoModel 类会根据 model 的类型自动选择合适的模型类。AutoTokenizer 会根据 model 的类型自动选择合适的分词器。AutoConfig 会根据 model 的类型自动选择合适的模型配置类。
在 transformers 中实现了大量的算法模型类,有 Bert 模型的 BertModel 类,有 BART 模型的 BartModel 类,有 GPT 模型的 GPT2Model 类等。
AutoTokenizer 会根据 model 的类型自动选择合适的分词器。需要注意的是,预训练模型与分词器是配套使用的。如果使用了 cardiffnlp/twitter-roberta-base-sentiment-latest 模型,就应该使用 cardiffnlp/twitter-roberta-base-sentiment-latest 分词器,否则效果会很差。
| |
本篇主要是介绍了 transformers 中的 pipeline、模型类、分词器、模型配置类等。pipeline 提供了便捷的方式,将分词器、模型处理、后处理器等组合在一起,方便用户使用。AutoModel 类会根据 model 的类型自动选择合适的模型类。AutoTokenizer 会根据 model 的类型自动选择合适的分词器。AutoConfig 会根据 model 的类型自动选择合适的模型配置类。