update __all__ export
parent
a950ec42b0
commit
bec1289fcd
ppcls/arch/backbone/model_zoo
|
@ -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(
|
||||
|
|
|
@ -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__()
|
||||
|
|
|
@ -16,6 +16,9 @@ MODEL_URLS = {
|
|||
}
|
||||
|
||||
|
||||
__all__ = MODEL_URLS.keys()
|
||||
|
||||
|
||||
trunc_normal_ = TruncatedNormal(std=.02)
|
||||
zeros_ = Constant(value=0.)
|
||||
ones_ = Constant(value=1.)
|
||||
|
|
Loading…
Reference in New Issue