提交 99af29e3 编写于 作者: X xuwei06

Fix error message for img_pool_layer

上级 d7b80f03
...@@ -2589,7 +2589,7 @@ def img_pool_layer(input, ...@@ -2589,7 +2589,7 @@ def img_pool_layer(input,
assert type(pool_type) in [AvgPooling, MaxPooling, CudnnAvgPooling, assert type(pool_type) in [AvgPooling, MaxPooling, CudnnAvgPooling,
CudnnMaxPooling], \ CudnnMaxPooling], \
"only AvgPooling and MaxPooling are supported" "only (Cudnn)AvgPooling, (Cudnn)MaxPooling are supported"
if pool_type is None: if pool_type is None:
pool_type = MaxPooling() pool_type = MaxPooling()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册