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