提交 a70087d7 编写于 作者: X xiaosang 提交者: Jinhua Liang

bug fix yolov3 (#3925)

上级 90019627
...@@ -189,10 +189,10 @@ def train(): ...@@ -189,10 +189,10 @@ def train():
sys.stdout.flush() sys.stdout.flush()
#add profiler tools #add profiler tools
if args.is_profiler and iter_id == 5: if args.is_profiler and iter_id == 5:
profiler.start_profiler("All") profiler.start_profiler("All")
elif args.is_profiler and iter_id == 10: elif args.is_profiler and iter_id == 10:
profiler.stop_profiler("total", args.profiler_path) profiler.stop_profiler("total", args.profiler_path)
return return
if (iter_id + 1) % cfg.snapshot_iter == 0: if (iter_id + 1) % cfg.snapshot_iter == 0:
save_model("model_iter{}".format(iter_id)) save_model("model_iter{}".format(iter_id))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册