From 90019627bbb66caf9f4c70457da7e2fa89a7cdb8 Mon Sep 17 00:00:00 2001 From: xiaosang <603336339@qq.com> Date: Wed, 13 Nov 2019 22:04:16 +0800 Subject: [PATCH] bug fix (#3924) --- PaddleCV/PaddleDetection/tools/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleCV/PaddleDetection/tools/train.py b/PaddleCV/PaddleDetection/tools/train.py index 166034fb..c688fef4 100644 --- a/PaddleCV/PaddleDetection/tools/train.py +++ b/PaddleCV/PaddleDetection/tools/train.py @@ -264,7 +264,7 @@ def main(): profiler.start_profiler("All") elif FLAGS.is_profiler and it == 10: profiler.stop_profiler("total", FLAGS.profiler_path) - return + return if (it > 0 and it % cfg.snapshot_iter == 0 or it == cfg.max_iters - 1) \ and (not FLAGS.dist or trainer_id == 0): -- GitLab