提交 ba85e189 编写于 作者: S SunAhong1993

fix the bug

上级 3c80edf9
......@@ -21,14 +21,14 @@ import numpy as np
class Decoder(object):
def _optimize_graph(self, graph):
torch._C._jit_pass_constant_propagation(graph)
# torch._C._jit_pass_dce(graph)
# torch._C._jit_pass_lint(graph)
# torch._C._jit_pass_peephole(graph)
# torch._C._jit_pass_lint(graph)
# torch._C._jit_pass_dce(graph)
# torch._C._jit_pass_lint(graph)
# torch._C._jit_pass_canonicalize(graph)
# torch._C._jit_pass_lint(graph)
torch._C._jit_pass_dce(graph)
torch._C._jit_pass_lint(graph)
torch._C._jit_pass_peephole(graph)
torch._C._jit_pass_lint(graph)
torch._C._jit_pass_dce(graph)
torch._C._jit_pass_lint(graph)
torch._C._jit_pass_canonicalize(graph)
torch._C._jit_pass_lint(graph)
torch._C._jit_pass_constant_propagation(graph)
return graph
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册