Merge pull request #4076 from chengduoZH/enable_cudnn_convt

add cudnn_convt layer type in config_helper.

Merge pull request #4076 from chengduoZH/enable_cudnn_convt
add cudnn_convt layer type in config_helper.
aed51dde · Cao Ying · GitHub · 47975870 · 22de57f1 · aed51dde
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

python/paddle/trainer_config_helpers/layers.py python/paddle/trainer_config_helpers/layers.py +1 -0

未找到文件。
--- 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'