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

Tell the limitation of the pooling on the GPU.

上级 0c6e0946
...@@ -80,13 +80,16 @@ def max_pool_2d(input, ds, ignore_border=None, st=None, padding=(0, 0), ...@@ -80,13 +80,16 @@ def max_pool_2d(input, ds, ignore_border=None, st=None, padding=(0, 0),
if input.ndim < 2: if input.ndim < 2:
raise NotImplementedError('max_pool_2d requires a dimension >= 2') raise NotImplementedError('max_pool_2d requires a dimension >= 2')
if ignore_border is None: if ignore_border is None:
warnings.warn("max_pool_2d() will have the parameter ignore_border" warnings.warn(
" default value changed to True (currently" "max_pool_2d() will have the parameter ignore_border"
" False). To have consistent behavior with all Theano" " default value changed to True (currently"
" version, explicitly add the parameter" " False). To have consistent behavior with all Theano"
" ignore_border=True. (this is also faster than" " version, explicitly add the parameter ignore_border=True."
" ignore_border=False)", " On the GPU, using ignore_border=False allow to use cudnn."
stacklevel=2) " Without ignore_border=True and cudnn, the only GPU combination"
" supported is when the `ds == st and padding == (0, 0) and"
" mode == 'max'`",
stacklevel=2)
ignore_border = False ignore_border = False
if input.ndim == 4: if input.ndim == 4:
op = DownsampleFactorMax(ds, ignore_border, st=st, padding=padding, op = DownsampleFactorMax(ds, ignore_border, st=st, padding=padding,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论