mindformers
Auto类
这是一个通用的配置类,当使用 from_pretrained() 类方法时,它会自动实例化模型Config类,并返回。 |
|
这是一个通用的模型类,使用类方法 AutoModel.from_pretrained 或 AutoModel.from_config 创建时会自动实例化为库中的基础模型类之一。 |
|
这是一个通用的模型类,使用类方法 AutoModelForCausalLM.from_pretrained 或 AutoModelForCausalLM.from_config 创建时会自动实例化为库中(带有因果语言建模头)的模型类之一。 |
|
这是一个通用的Processor类,当使用 from_pretrained() 类方法时,它会自动实例化Processor类,并返回。 |
|
这是一个通用的Tokenizer类,当使用 from_pretrained() 类方法时,它会自动实例化Tokenzier类,并返回。 |
Trainer高阶接口
Trainer是通用的任务执行组件,通过参数中的task名称或者配置文件可以指定任务并且初始化一个与该任务相关的trainer实例。 |
|
TrainingArguments是与MindSpore训练相关的参数集合。 |
pipeline高阶接口
通过流水线执行套件中已集成任务和模型的推理流程。 |
运行检查
检查CANN、固件与驱动、MindSpore和MindFormers安装版本是否匹配。 |
面向服务化
用于将 MindFormers 的一个模型运行实例作为 MindIEServer 的后端。 |