diff --git a/ppcls/modeling/architectures/ghostnet.py b/ppcls/modeling/architectures/ghostnet.py index 79dd918966f012bb5b80467b52d6e08a7f709159..ace89a93a6e9366c5f8b79b07af12bcec451f417 100644 --- a/ppcls/modeling/architectures/ghostnet.py +++ b/ppcls/modeling/architectures/ghostnet.py @@ -21,6 +21,8 @@ from paddle.nn import Conv2D, BatchNorm, AdaptiveAvgPool2D, Linear from paddle.regularizer import L2Decay from paddle.nn.initializer import Uniform, KaimingNormal +__all__ = ["GhostNet_x0_5", "GhostNet_x1_0", "GhostNet_x1_3"] + class ConvBNLayer(nn.Layer): def __init__(self,