def __init__()

in containers/Shoot/CNN/unet.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, num_filter,kernel,pad,activation,**kwargs):
        super(Conv_Block, self).__init__(**kwargs)
        with self.name_scope():
            self.conv=nn.Conv2D(num_filter,kernel,padding=pad)
            self.norm=nn.BatchNorm()
            if activation=="swish":
                self.act=nn.Swish()
            else:
                self.act=nn.Activation(activation)