in containers/Shoot/CNN/unet.py [0:0]
def __init__(self, depth=5,activation="relu",width=8,**kwargs):
super(Up_Branch, self).__init__(**kwargs)
kernel=(3,3)
pad=(1,1)
with self.name_scope():
self.blocks=[
Up_Block(pow(2,width-i),kernel=3,pad=1,activation=activation)
for i in range(depth)]
[self.register_child(x) for x in self.blocks]
self.conv=nn.Conv2D(1,kernel_size=1)