def __init__(self, *args, **kwargs):
FuseNet.__init__(self, *args, **kwargs)
self.set_blocks(block_depth=4)
def __init__(self, skip=3, *args, **kwargs):
FuseNet.__init__(self, *args, **kwargs)
self.block_depth = (self.num_depth - 2) / skip
self.set_blocks(block_depth=self.block_depth)