提交 f8490a2c 编写于 作者: Q Qiao Longfei

add sync mode option

上级 87c5cf65
...@@ -92,6 +92,8 @@ def parse_args(): ...@@ -92,6 +92,8 @@ def parse_args():
default='GPU', default='GPU',
choices=['CPU', 'GPU'], choices=['CPU', 'GPU'],
help="The device type.") help="The device type.")
parser.add_argument(
'--sync', type=ast.literal_eval, default=True, help="sync mode.")
args = parser.parse_args() args = parser.parse_args()
# Append args related to dict # Append args related to dict
...@@ -256,6 +258,9 @@ def train(args): ...@@ -256,6 +258,9 @@ def train(args):
beta2=TrainTaskConfig.beta2, beta2=TrainTaskConfig.beta2,
epsilon=TrainTaskConfig.eps) epsilon=TrainTaskConfig.eps)
optimizer.minimize(sum_cost) optimizer.minimize(sum_cost)
elif args.sync == False:
optimizer = fluid.optimizer.SGD(0.003)
optimizer.minimize(sum_cost)
else: else:
lr_decay = fluid.layers\ lr_decay = fluid.layers\
.learning_rate_scheduler\ .learning_rate_scheduler\
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册