Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
提交 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.
先完成此消息的编辑！
想要评论请 注册 或