mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix paddle2.4 hang problem

pull/2648/head
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 compiled_train_prog = train_prog
if eval_dataloader is not None: 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"]): for epoch_id in range(global_config["epochs"]):
# 1. train with train dataset # 1. train with train dataset