fix paddle2.4 hang problem

2023-02-13 13:22:20 +00:00 · 2023-02-13 13:22:20 +00:00 · 85f65ce76f
parent 3f43784964
commit 85f65ce76f
1 changed files with 4 additions and 1 deletions
--- a/ppcls/static/train.py
+++ b/ppcls/static/train.py
@ -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