in containers/Shoot/CNN/cnn.py [0:0]
def __init__(self, num_filter,kernel,pad,**kwargs):
super(Block, self).__init__()
with self.name_scope():
self.conv_1=nn.Conv2D(num_filter,kernel,padding=pad,groups=4)
self.conv_2=nn.Conv2D(num_filter,kernel,padding=pad,groups=2)
self.norm_1=nn.BatchNorm()
self.norm_2=nn.BatchNorm()
self.act_1=nn.Activation("relu")
self.act_2=nn.Activation("relu")