Browse Source

enable fused_attn and fused_linear for CLIP_vit series (#3743)

Shuhao Liang 6 months ago
parent
commit
a6c7e205a2

+ 2 - 0
paddlex/repo_apis/PaddleClas_api/configs/CLIP_vit_base_patch16_224.yaml

@@ -30,6 +30,8 @@ Arch:
   class_num: 1000
   return_embed: False
   pretrained: True
+  use_fused_attn: True
+  use_fused_linear: True
 
 # loss function config for traing/eval process
 Loss:

+ 2 - 0
paddlex/repo_apis/PaddleClas_api/configs/CLIP_vit_large_patch14_224.yaml

@@ -27,6 +27,8 @@ Arch:
   class_num: 1000
   return_embed: False
   pretrained: True
+  use_fused_attn: True
+  use_fused_linear: True
 
 # loss function config for traing/eval process
 Loss: