in containers/Shoot/CNN/unet.py [0:0]
def __init__(self, activation,depth=5,pool='avg',width=5,**kwargs):
super(Down_Branch, self).__init__(**kwargs)
kernel=(3,3)
pad=(1,1)
with self.name_scope():
self.blocks=[
Down_Block(pow(2,width+i),kernel=kernel,pad=pad,pool=(i!=depth-1),activation=activation)
for i in range(depth)]
[self.register_child(x) for x in self.blocks]