cleanup enum test=develop (#29294)

revert-31562-mean
gongweibao 5 years ago committed by GitHub
parent 696dc4bb13
commit 96de8b008f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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
"""

Loading…
Cancel
Save