mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix paddle2.4 hang problem

Browse Source
This commit is contained in:
kangguangli 2023-02-13 13:22:20 +00:00 committed by Wei Shengyu
parent 3f43784964
commit 85f65ce76f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/static/train.py
View File

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