diff --git a/PaddleCV/yolov3/train.py b/PaddleCV/yolov3/train.py index a89c6b30ea71804e017ad87eb0e6fef232c12de1..77ea678ff0cd10cf76bf904dcefa26ed0ae7e7c8 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))