提交 757f0d20 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Better error message when abstract conv does not get replaced

上级 e6d7fbb8
...@@ -303,7 +303,10 @@ class AbstractConv2d(BaseAbstractConv2d): ...@@ -303,7 +303,10 @@ class AbstractConv2d(BaseAbstractConv2d):
return Apply(self, [img, kern], [output]) return Apply(self, [img, kern], [output])
def perform(self, node, inp, out_): def perform(self, node, inp, out_):
raise NotImplementedError('AbstractConv2d theano optimization failed') raise NotImplementedError(
'AbstractConv2d theano optimization failed. '
'Did you exclude both "conv_dnn" and "conv_gemm" from '
'the optimizer?')
def grad(self, inp, grads): def grad(self, inp, grads):
bottom, weights = inp bottom, weights = inp
...@@ -378,7 +381,9 @@ class AbstractConv2d_gradWeights(BaseAbstractConv2d): ...@@ -378,7 +381,9 @@ class AbstractConv2d_gradWeights(BaseAbstractConv2d):
def perform(self, node, inp, out_): def perform(self, node, inp, out_):
raise NotImplementedError( raise NotImplementedError(
'AbstractConv2d_gradWeight theano optimization failed') 'AbstractConv2d_gradWeights theano optimization failed. '
'Did you exclude both "conv_dnn" and "conv_gemm" from '
'the optimizer?')
def grad(self, inp, grads): def grad(self, inp, grads):
bottom, top = inp[:2] bottom, top = inp[:2]
...@@ -457,7 +462,9 @@ class AbstractConv2d_gradInputs(BaseAbstractConv2d): ...@@ -457,7 +462,9 @@ class AbstractConv2d_gradInputs(BaseAbstractConv2d):
def perform(self, node, inp, out_): def perform(self, node, inp, out_):
raise NotImplementedError( raise NotImplementedError(
'AbstractConv2d_gradWeight theano optimization failed') 'AbstractConv2d_gradInputs theano optimization failed. '
'Did you exclude both "conv_dnn" and "conv_gemm" from '
'the optimizer?')
def grad(self, inp, grads): def grad(self, inp, grads):
weights, top = inp[:2] weights, top = inp[:2]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论