update __all__ export

pull/826/head
jm12138 2021-06-10 23:20:29 +08:00
parent a950ec42b0
commit bec1289fcd
3 changed files with 9 additions and 0 deletions
ppcls/arch/backbone/model_zoo

View File

@ -16,6 +16,9 @@ MODEL_URLS = {
}
__all__ = MODEL_URLS.keys()
def ConvLayer(in_channels, out_channels, kernel_size=3, stride=1, bias_attr=False):
layer = nn.Sequential(
('conv', nn.Conv2D(

View File

@ -20,6 +20,9 @@ MODEL_URLS = {
}
__all__ = MODEL_URLS.keys()
class Involution(nn.Layer):
def __init__(self, channels, kernel_size, stride):
super(Involution, self).__init__()

View File

@ -16,6 +16,9 @@ MODEL_URLS = {
}
__all__ = MODEL_URLS.keys()
trunc_normal_ = TruncatedNormal(std=.02)
zeros_ = Constant(value=0.)
ones_ = Constant(value=1.)