def init_taskd_worker(rank_id: int, upper_limit_of_disk_in_mb: int = 5000) -> bool

功能说明

用户侧代码调用此函数,初始化taskd worker。

输入参数说明

表1 输入参数说明

参数

类型

说明

rank_id

int

当前训练进程的global rank号。

upper_limit_of_disk_in_mb

int

所有训练进程能使用的profiling文件夹存储空间上限,实际大小在此阈值上下波动,单位为MB,非负值,默认5000。

返回值说明

返回值类型

说明

bool

表明初始化是否成功。

True:初始化成功。

False:初始化失败。