未验证 提交 96de8b00 编写于 作者: G gongweibao 提交者: GitHub

cleanup enum test=develop (#29294)

上级 696dc4bb
...@@ -22,7 +22,7 @@ def get_cloud_cluster(args_node_ips, ...@@ -22,7 +22,7 @@ def get_cloud_cluster(args_node_ips,
devices_per_proc, devices_per_proc,
args_port=6170): args_port=6170):
""" """
args_node_ips:string, device_mode:DeviceMode(IntEnum), device_per_proc:list, args_port: int args_node_ips:string, device_mode:DeviceMode(Int), device_per_proc:list, args_port: int
""" """
#you can automatically get ip info while using paddlecloud multi nodes mode. #you can automatically get ip info while using paddlecloud multi nodes mode.
node_ips = os.getenv("PADDLE_TRAINERS") node_ips = os.getenv("PADDLE_TRAINERS")
......
...@@ -27,7 +27,6 @@ from contextlib import closing ...@@ -27,7 +27,6 @@ from contextlib import closing
import socket import socket
import warnings import warnings
import six import six
from enum import IntEnum
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
...@@ -35,7 +34,7 @@ logger = logging.getLogger("root") ...@@ -35,7 +34,7 @@ logger = logging.getLogger("root")
logger.propagate = False logger.propagate = False
class DistributeMode(IntEnum): class DistributeMode():
""" """
There are various mode for fleetrun, each of them is designed for different model. There are various mode for fleetrun, each of them is designed for different model.
""" """
...@@ -44,7 +43,7 @@ class DistributeMode(IntEnum): ...@@ -44,7 +43,7 @@ class DistributeMode(IntEnum):
PS_HETER = 2 PS_HETER = 2
class DeviceMode(IntEnum): class DeviceMode():
""" """
Training devices type Training devices type
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册