fix save model bug when use distributed traning (#4533)

282e4890 · LielinJiang · GitHub · 269b1767 · 282e4890
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleCV/image_classification/train.py PaddleCV/image_classification/train.py +1 -1

未找到文件。
--- a/PaddleCV/image_classification/train.py
+++ b/PaddleCV/image_classification/train.py
@@ -288,7 +288,7 @@ def train(args):
            if args.use_dali:
                test_iter.reset()

-        if pass_id % args.save_step == 0:
+        if trainer_id == 0 and pass_id % args.save_step == 0:
            save_model(args, exe, train_prog, pass_id)