get_world_size

paddle.distributed. get_world_size ( ) [源代码]

返回参与当前任务的进程数。

当前进程数等于环境变量 PADDLE_TRAINERS_NUM 的值,默认值为1。

返回

(int) 参与任务的进程数。

代码示例

import paddle
import paddle.distributed as dist

# execute this command in terminal: export PADDLE_TRAINERS_NUM=4
print("The world_size is %d" % dist.get_world_size())
# The world_size is 4