Fix error message for img_pool_layer

8 years ago · 99af29e3f2
parent d7b80f03b0
commit 99af29e3f2
1 changed files with 4 additions and 4 deletions
--- a/python/paddle/trainer_config_helpers/layers.py
+++ b/python/paddle/trainer_config_helpers/layers.py
@ -2589,7 +2589,7 @@ def img_pool_layer(input,

    assert type(pool_type) in [AvgPooling, MaxPooling, CudnnAvgPooling,
                               CudnnMaxPooling], \
-        "only AvgPooling and MaxPooling are supported"
+        "only (Cudnn)AvgPooling, (Cudnn)MaxPooling are supported"

    if pool_type is None:
        pool_type = MaxPooling()