昇腾社区首页
中文
注册

def init_taskd_agent(config : dict = {}, cls = None) -> bool

功能说明

用户侧调用此函数,初始化TaskD Agent。

输入参数说明

表1 参数说明

参数

类型

说明

config

dict:{str : str}

Agent配置信息,包括Agent配置与网络配置。

其中键包括:

  • Framework:Agent框架,当前支持PyTorch和MindSpore
  • UpstreamAddr:网络侧上游IP地址
  • UpstreamPort:网络侧上游端口
  • ServerRank:Agent rank号

cls

具体实例类型

该入参在PyTorch框架下使用,为SimpleElasticAgent实例。其他框架无需传入。

返回值说明

表2 返回值说明

返回值类型

说明

bool

表明初始化是否成功。

  • True:初始化成功。
  • False:初始化失败。