提交 ca3529d2 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

fixed another except TypeError

上级 2d52b5f1
...@@ -1165,13 +1165,13 @@ class ScanMerge(gof.Optimizer): ...@@ -1165,13 +1165,13 @@ class ScanMerge(gof.Optimizer):
nsteps = node.inputs[0] nsteps = node.inputs[0]
try: try:
nsteps = int(get_scalar_constant_value(nsteps)) nsteps = int(get_scalar_constant_value(nsteps))
except TypeError: except tensor.NotScalarConstantError:
pass pass
rep_nsteps = rep.inputs[0] rep_nsteps = rep.inputs[0]
try: try:
rep_nsteps = int(get_scalar_constant_value(rep_nsteps)) rep_nsteps = int(get_scalar_constant_value(rep_nsteps))
except TypeError: except tensor.NotScalarConstantError:
pass pass
# Check to see if it is an input of a different node # Check to see if it is an input of a different node
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论