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

cleanup enum test=develop (#29294)

上级 696dc4bb
......@@ -22,7 +22,7 @@ def get_cloud_cluster(args_node_ips,
devices_per_proc,
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.
node_ips = os.getenv("PADDLE_TRAINERS")
......
......@@ -27,7 +27,6 @@ from contextlib import closing
import socket
import warnings
import six
from enum import IntEnum
import paddle
import paddle.fluid as fluid
......@@ -35,7 +34,7 @@ logger = logging.getLogger("root")
logger.propagate = False
class DistributeMode(IntEnum):
class DistributeMode():
"""
There are various mode for fleetrun, each of them is designed for different model.
"""
......@@ -44,7 +43,7 @@ class DistributeMode(IntEnum):
PS_HETER = 2
class DeviceMode(IntEnum):
class DeviceMode():
"""
Training devices type
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册