Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 22de57f1 编写于 作者: C chengduoZH
浏览文件
操作

enable cudnn_convt

上级 e4bab9a4
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
python/paddle/trainer_config_helpers/layers.py
浏览文件 @ 22de57f1
...@@ -169,6 +169,7 @@ class LayerType(object): ...@@ -169,6 +169,7 @@ class LayerType(object):
EXCONV_LAYER = 'exconv' EXCONV_LAYER = 'exconv'
EXCONVTRANS_LAYER = 'exconvt' EXCONVTRANS_LAYER = 'exconvt'
CUDNNCONV_LAYER = 'cudnn_conv' CUDNNCONV_LAYER = 'cudnn_conv'
CUDNNCONVTRANS_LAYER = 'cudnn_convt'
POOL_LAYER = 'pool' POOL_LAYER = 'pool'
POOL3D_LAYER = 'pool3d' POOL3D_LAYER = 'pool3d'
BATCH_NORM_LAYER = 'batch_norm' BATCH_NORM_LAYER = 'batch_norm'
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或