get_world_size

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

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

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

返回

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

代码示例

  1. import paddle
  2. import paddle.distributed as dist
  3. # execute this command in terminal: export PADDLE_TRAINERS_NUM=4
  4. print("The world_size is %d" % dist.get_world_size())
  5. # The world_size is 4