From a70087d7249b0993b1c88cf6e1e8386f8d596320 Mon Sep 17 00:00:00 2001
From: xiaosang <603336339@qq.com>
Date: Wed, 13 Nov 2019 22:11:38 +0800
Subject: [PATCH] bug fix yolov3 (#3925)

---
 PaddleCV/yolov3/train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/PaddleCV/yolov3/train.py b/PaddleCV/yolov3/train.py
index a89c6b30..77ea678f 100644
--- a/PaddleCV/yolov3/train.py
+++ b/PaddleCV/yolov3/train.py
@@ -189,10 +189,10 @@ def train():
             sys.stdout.flush()
             #add profiler tools
             if args.is_profiler and iter_id == 5:
-               profiler.start_profiler("All")
+                profiler.start_profiler("All")
             elif args.is_profiler and iter_id == 10:
-                 profiler.stop_profiler("total", args.profiler_path)
-                 return
+                profiler.stop_profiler("total", args.profiler_path)
+                return
 
             if (iter_id + 1) % cfg.snapshot_iter == 0:
                 save_model("model_iter{}".format(iter_id))
-- 
GitLab