in containers/Shoot/CNN/cnn.py [0:0]
def __init__(self, width=16,depth=8,**kwargs):
super(CNN, self).__init__()
with self.name_scope():
self.conv_first=nn.Conv2D(width,1,padding=0)
self.conv=[Block(width,3,1) for x in range(depth)]
[self.register_child(x) for x in self.conv]
self.conv_last=nn.Conv2D(1,1,padding=0)
self.last_act=nn.Activation("sigmoid")