mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ppcls/arch/backbone/model_zoo/hardnet.py
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): def ConvLayer(in_channels, out_channels, kernel_size=3, stride=1, bias_attr=False):
layer = nn.Sequential( layer = nn.Sequential(
('conv', nn.Conv2D( ('conv', nn.Conv2D(

3
ppcls/arch/backbone/model_zoo/rednet.py
View File

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

3
ppcls/arch/backbone/model_zoo/tnt.py
View File

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