提交 dbb49e4b authored 作者: Frederic Bastien's avatar Frederic Bastien

Better error message

上级 12c4530d
...@@ -1143,6 +1143,8 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM): ...@@ -1143,6 +1143,8 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM):
raise ValueError('shape must be given if subsample != (1, 1)' raise ValueError('shape must be given if subsample != (1, 1)'
' or border_mode == "half"') ' or border_mode == "half"')
height_width = [shape[0], shape[1]] height_width = [shape[0], shape[1]]
assert shape[0].ndim == 0
assert shape[1].ndim == 0
else: else:
height_width = [] height_width = []
...@@ -1201,6 +1203,9 @@ class GpuCorrMM_gradInputs(BaseGpuCorrMM): ...@@ -1201,6 +1203,9 @@ class GpuCorrMM_gradInputs(BaseGpuCorrMM):
if self.subsample != (1, 1) and shape is None: if self.subsample != (1, 1) and shape is None:
raise ValueError('shape must be given if subsample != (1, 1)') raise ValueError('shape must be given if subsample != (1, 1)')
height_width = [shape[0], shape[1]] if self.subsample != (1, 1) else [] height_width = [shape[0], shape[1]] if self.subsample != (1, 1) else []
if height_width:
assert shape[0].ndim == 0
assert shape[1].ndim == 0
broadcastable = [topgrad.type.broadcastable[0], kern.type.broadcastable[1], broadcastable = [topgrad.type.broadcastable[0], kern.type.broadcastable[1],
False, False] False, False]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论