Merge pull request #4076 from chengduoZH/enable_cudnn_convt

add cudnn_convt layer type in config_helper.
8 years ago · aed51dde32
parent 47975870aa 22de57f127
commit aed51dde32
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/trainer_config_helpers/layers.py
+++ b/python/paddle/trainer_config_helpers/layers.py
@ -169,6 +169,7 @@ class LayerType(object):
    EXCONV_LAYER = 'exconv'
    EXCONVTRANS_LAYER = 'exconvt'
    CUDNNCONV_LAYER = 'cudnn_conv'
+    CUDNNCONVTRANS_LAYER = 'cudnn_convt'
    POOL_LAYER = 'pool'
    POOL3D_LAYER = 'pool3d'
    BATCH_NORM_LAYER = 'batch_norm'