提交 c2a9f883 编写于 作者: D dongshuilong

update profiler

上级 272bc948
......@@ -24,7 +24,7 @@ def train_epoch(engine, epoch_id, print_batch_step):
for iter_id, batch in enumerate(engine.train_dataloader):
if iter_id >= engine.max_iter:
break
profiler.add_profiler_step(trainer.config["profiler_options"])
profiler.add_profiler_step(engine.config["profiler_options"])
if iter_id == 5:
for key in engine.time_info:
engine.time_info[key].reset()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册