Add checks for the length of outputs in CTC's L_op function

上级 ed282869
......@@ -117,6 +117,7 @@ class GpuConnectionistTemporalClassification(gof.COp):
def L_op(self, inputs, outputs, output_grads):
# Gradients computed by Op
assert self.compute_grad and len(outputs) == 2
gradients = outputs[1]
assert gradients is not None
......
......@@ -178,6 +178,7 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp):
outputs=outputs)
def L_op(self, inputs, outputs, output_grads):
assert self.compute_grad and len(outputs) == 2
gradients = outputs[1]
assert gradients is not None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论