diff --git a/ppcls/modeling/architectures/__init__.py b/ppcls/modeling/architectures/__init__.py
index 6ab6e67146e19f81a5918e7ed0d9955dd067b44a..ac57a786aa16dcdda9c418a130fd4423c721e421 100644
--- a/ppcls/modeling/architectures/__init__.py
+++ b/ppcls/modeling/architectures/__init__.py
@@ -46,4 +46,4 @@ from .resnet_acnet import ResNet18_ACNet, ResNet34_ACNet, ResNet50_ACNet, ResNet
 # distillation model
 from .distillation_models import ResNet50_vd_distill_MobileNetV3_large_x1_0, ResNeXt101_32x16d_wsl_distill_ResNet50_vd
 
-from .csp_resnet import CSPResNet50, CSPResNet101
\ No newline at end of file
+from .csp_resnet import CSPResNet50_leaky
\ No newline at end of file