提交 bec1289f 编写于 作者: jm_12138's avatar jm_12138

update __all__ export

上级 a950ec42
...@@ -16,6 +16,9 @@ MODEL_URLS = { ...@@ -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(
......
...@@ -20,6 +20,9 @@ MODEL_URLS = { ...@@ -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__()
......
...@@ -16,6 +16,9 @@ MODEL_URLS = { ...@@ -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.)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册