提交 4a20e672 编写于 作者: G guosheng

Refine the profiler in Transformer

上级 10b1378d
...@@ -130,6 +130,8 @@ def parse_args(): ...@@ -130,6 +130,8 @@ def parse_args():
def main(args): def main(args):
train_prog = fluid.Program() train_prog = fluid.Program()
startup_prog = fluid.Program() startup_prog = fluid.Program()
train_prog.random_seed = 1000
startup_prog.random_seed = 1000
with fluid.program_guard(train_prog, startup_prog): with fluid.program_guard(train_prog, startup_prog):
with fluid.unique_name.guard(): with fluid.unique_name.guard():
sum_cost, avg_cost, predict, token_num, pyreader = transformer( sum_cost, avg_cost, predict, token_num, pyreader = transformer(
...@@ -248,7 +250,6 @@ def main(args): ...@@ -248,7 +250,6 @@ def main(args):
if args.use_py_reader: if args.use_py_reader:
pyreader.reset() pyreader.reset()
pyreader.start() pyreader.start()
break
return reader_time, run_time return reader_time, run_time
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册