2318052dc79966bf36675606b7d992a347418292,gluoncv/model_zoo/nasnet.py,BranchSeparables,__init__,#BranchSeparables#Any#Any#Any#Any#Any#Any#,75

Before Change

        super(BranchSeparables, self).__init__()
        self.body = nn.HybridSequential(prefix="")

        self.body.add(nn.Activation("relu"))
        self.body.add(SeparableConv2d(in_channels, in_channels, kernel_size,
                                      stride, padding, use_bias=use_bias))
        self.body.add(nn.BatchNorm(momentum=0.1, epsilon=0.001))
        self.body.add(nn.Activation("relu"))

After Change

    def __init__(self, in_channels, out_channels, kernel_size, stride, padding,
                 norm_layer, norm_kwargs, use_bias=False):
        super(BranchSeparables, self).__init__()
        self.body = nn.HybridSequential(prefix="")

        self.body.add(nn.Activation("relu"))
        self.body.add(SeparableConv2d(in_channels, in_channels, kernel_size,
                                      stride, padding, use_bias=use_bias))
        self.body.add(norm_layer(momentum=0.1, epsilon=0.001,
                                 **({} if norm_kwargs is None else norm_kwargs)))

Instances