def init_taskd_worker(rank_id: int, upper_limit_of_disk_in_mb: int = 5000) -> bool
功能说明
用户侧代码调用此函数,初始化taskd worker。
输入参数说明
参数 |
类型 |
说明 |
---|---|---|
rank_id |
int |
当前训练进程的global rank号。 |
upper_limit_of_disk_in_mb |
int |
所有训练进程能使用的profiling文件夹存储空间上限,实际大小在此阈值上下波动,单位为MB,非负值,默认5000。 |
返回值说明
返回值类型 |
说明 |
---|---|
bool |
表明初始化是否成功。 True:初始化成功。 False:初始化失败。 |
父主题: TaskD