def init_taskd_worker(rank_id: int, upper_limit_of_disk_in_mb: int = 5000, framework: str = "pt") -> bool
Description
Initializes TaskD Worker. This API is called in user code.
Input Parameters
Parameter |
Type |
Description |
|---|---|---|
rank_id |
Integer |
Global rank of the current training process. |
upper_limit_of_disk_in_mb |
Integer |
Maximum size (MB) of the Profiling folder that can be used by all training processes. The actual size fluctuates around the threshold. The value is a non-negative number. The default value is 5000. |
framework |
String |
AI framework used by a job |
Return Value
Type |
Description |
|---|---|
Bool |
Whether initialization is successful.
|
Parent topic: TaskD Worker APIs