昇腾社区首页
中文
注册

set_mindio_export_version

接口功能

设置框架类型和版本。

接口格式

mindio_ttp.set_mindio_export_version(framework_type: FrameworkType, version: int = 0)

接口参数

参数

是否必选

说明

取值要求

framework_type

必选

框架类型。

  • MODELLINK:ModelLink框架
  • HULK:Hulk框架
  • FrameworkType.MODELLINK=0
  • FrameworkType.HULK=1

version

可选

区分ModelLlink框架版本。

默认0

  • 0:对应ModelLlink目前最新支持Megatron-LM的core_r0.6.0分支。
  • 1:对应ModelLlink的tag:0.1.0支持Megatron-LM的23.06分支。
  • 2:对应Megatron-LM模型的commit id:285068。

返回值