提交 9c469831 编写于 作者: S ShawnXuan

12315->50051

上级 7feaa800
......@@ -52,7 +52,7 @@ def get_parser(parser=None):
help='node/machine number for training')
parser.add_argument('--node_ips', type=str_list, default=['192.168.1.13', '192.168.1.14'],
help='nodes ip list for training, devided by ",", length >= num_nodes')
parser.add_argument("--ctrl_port", type=int, default=12315, help='ctrl_port for multinode job')
parser.add_argument("--ctrl_port", type=int, default=50051, help='ctrl_port for multinode job')
parser.add_argument("--model", type=str, default="resnet50",
help="resnet50")
......
......@@ -50,7 +50,7 @@ parser.add_argument('--num_nodes', type=int, default=1,
help='node/machine number for training')
parser.add_argument('--node_ips', type=str_list, default=['192.168.1.13', '192.168.1.14'],
help='nodes ip list for training, devided by ",", length >= num_nodes')
parser.add_argument("--ctrl_port", type=int, default=12315, help='ctrl_port for multinode job')
parser.add_argument("--ctrl_port", type=int, default=50051, help='ctrl_port for multinode job')
parser.add_argument('--hidden_units_num', type=int, default=7)
parser.add_argument('--hidden_size', type=int, default=1024)
......
......@@ -48,7 +48,7 @@ def get_parser(parser=None):
help='node/machine number for training')
parser.add_argument('--node_ips', type=str_list, default=['192.168.1.13', '192.168.1.14'],
help='nodes ip list for training, devided by ",", length >= num_nodes')
parser.add_argument("--ctrl_port", type=int, default=12315, help='ctrl_port for multinode job')
parser.add_argument("--ctrl_port", type=int, default=50051, help='ctrl_port for multinode job')
# train
parser.add_argument("--learning_rate", type=float, default=1e-4, help="Learning rate")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册