提交 7fbece47 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Small fixes.

上级 6286ae0c
...@@ -1777,9 +1777,6 @@ class GpuCorr3dMM(BaseGpuCorr3dMM): ...@@ -1777,9 +1777,6 @@ class GpuCorr3dMM(BaseGpuCorr3dMM):
Parameters Parameters
---------- ----------
border_mode border_mode
Currently supports "valid" only; "full" can be simulated by setting
`pad="full"` (at the cost of performance), or by using
`GpuCorrMM_gradInputs`.
The width of a border of implicit zeros to pad the The width of a border of implicit zeros to pad the
input with. Must be a tuple with 3 elements giving the width of input with. Must be a tuple with 3 elements giving the width of
the padding on each side, or a single integer to pad the same the padding on each side, or a single integer to pad the same
......
...@@ -974,7 +974,7 @@ class BaseAbstractConv(Op): ...@@ -974,7 +974,7 @@ class BaseAbstractConv(Op):
if len(border_mode) != convdim: if len(border_mode) != convdim:
raise ValueError( raise ValueError(
'border mode must have exactly {} values, ' 'border mode must have exactly {} values, '
'but was {}'.format(convdim)) 'but was {}'.format(convdim, border_mode))
border_mode = tuple(map(int, border_mode)) border_mode = tuple(map(int, border_mode))
if border_mode == (0,) * convdim: if border_mode == (0,) * convdim:
border_mode = 'valid' border_mode = 'valid'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论