提交 6286ae0c authored 作者: Gijs van Tulder's avatar Gijs van Tulder

GpuCorrMM and GpuCorr3dMM have default pad=None.

上级 c1df889d
...@@ -865,8 +865,8 @@ class BaseGpuCorrMM(GpuOp): ...@@ -865,8 +865,8 @@ class BaseGpuCorrMM(GpuOp):
__props__ = ('border_mode', 'subsample', 'filter_dilation') __props__ = ('border_mode', 'subsample', 'filter_dilation')
def __init__(self, border_mode="valid", subsample=(1, 1), def __init__(self, border_mode="valid", subsample=(1, 1),
filter_dilation=(1, 1), pad=(0, 0)): filter_dilation=(1, 1), pad=None):
if pad != (0, 0): if pad is not None:
_logger.warning( _logger.warning(
'do not use pad for BaseGpuCorrMM; please set padding in ' 'do not use pad for BaseGpuCorrMM; please set padding in '
'border_mode parameter, see the docstring for more details') 'border_mode parameter, see the docstring for more details')
...@@ -1267,7 +1267,7 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM): ...@@ -1267,7 +1267,7 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM):
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1), subsample=(1, 1),
filter_dilation=(1, 1), filter_dilation=(1, 1),
pad=(0, 0)): pad=None):
super(GpuCorrMM_gradWeights, self).__init__(border_mode, super(GpuCorrMM_gradWeights, self).__init__(border_mode,
subsample, subsample,
filter_dilation, filter_dilation,
...@@ -1338,7 +1338,7 @@ class GpuCorrMM_gradInputs(BaseGpuCorrMM): ...@@ -1338,7 +1338,7 @@ class GpuCorrMM_gradInputs(BaseGpuCorrMM):
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1), subsample=(1, 1),
filter_dilation=(1, 1), filter_dilation=(1, 1),
pad=(0, 0)): pad=None):
super(GpuCorrMM_gradInputs, self).__init__(border_mode, subsample, super(GpuCorrMM_gradInputs, self).__init__(border_mode, subsample,
filter_dilation, pad) filter_dilation, pad)
...@@ -1415,8 +1415,8 @@ class BaseGpuCorr3dMM(GpuOp): ...@@ -1415,8 +1415,8 @@ class BaseGpuCorr3dMM(GpuOp):
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1, 1), subsample=(1, 1, 1),
filter_dilation=(1, 1, 1), filter_dilation=(1, 1, 1),
pad=(0, 0, 0)): pad=None):
if pad != (0, 0, 0): if pad is not None:
_logger.warning( _logger.warning(
'do not use pad for BaseGpuCorr3dMM; please set padding in ' 'do not use pad for BaseGpuCorr3dMM; please set padding in '
'border_mode parameter, see the docstring for more details') 'border_mode parameter, see the docstring for more details')
...@@ -1821,7 +1821,7 @@ class GpuCorr3dMM(BaseGpuCorr3dMM): ...@@ -1821,7 +1821,7 @@ class GpuCorr3dMM(BaseGpuCorr3dMM):
""" """
def __init__(self, border_mode="valid", subsample=(1, 1, 1), def __init__(self, border_mode="valid", subsample=(1, 1, 1),
filter_dilation=(1, 1, 1), pad=(0, 0, 0)): filter_dilation=(1, 1, 1), pad=None):
super(GpuCorr3dMM, self).__init__(border_mode, subsample, super(GpuCorr3dMM, self).__init__(border_mode, subsample,
filter_dilation, pad) filter_dilation, pad)
...@@ -1871,7 +1871,7 @@ class GpuCorr3dMM_gradWeights(BaseGpuCorr3dMM): ...@@ -1871,7 +1871,7 @@ class GpuCorr3dMM_gradWeights(BaseGpuCorr3dMM):
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1, 1), subsample=(1, 1, 1),
filter_dilation=(1, 1, 1), filter_dilation=(1, 1, 1),
pad=(0, 0, 0)): pad=None):
super(GpuCorr3dMM_gradWeights, self).__init__(border_mode, subsample, super(GpuCorr3dMM_gradWeights, self).__init__(border_mode, subsample,
filter_dilation, pad) filter_dilation, pad)
...@@ -1941,7 +1941,7 @@ class GpuCorr3dMM_gradInputs(BaseGpuCorr3dMM): ...@@ -1941,7 +1941,7 @@ class GpuCorr3dMM_gradInputs(BaseGpuCorr3dMM):
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1, 1), subsample=(1, 1, 1),
filter_dilation=(1, 1, 1), filter_dilation=(1, 1, 1),
pad=(0, 0, 0)): pad=None):
super(GpuCorr3dMM_gradInputs, self).__init__(border_mode, subsample, super(GpuCorr3dMM_gradInputs, self).__init__(border_mode, subsample,
filter_dilation, pad) filter_dilation, pad)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论