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

update __all__ export

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