提交 85f65ce7 编写于 作者: K kangguangli 提交者: Wei Shengyu

fix paddle2.4 hang problem

上级 3f437849
......@@ -171,7 +171,10 @@ def main(args):
compiled_train_prog = train_prog
if eval_dataloader is not None:
compiled_eval_prog = program.compile(config, eval_prog)
if not global_config.get("is_distributed", True):
compiled_eval_prog = program.compile(config, eval_prog)
else:
compiled_eval_prog = eval_prog
for epoch_id in range(global_config["epochs"]):
# 1. train with train dataset
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册