def __init__(self, channels, norm_type='bn'): Model.__init__(self) self.channels = channels self.norm_type = norm_type
def __init__(self, classes, norm_type='bn'): Model.__init__(self) self.classes = classes self.norm_type = norm_type
def __init__(self, norm_type='bn'): Model.__init__(self) self.norm_type = norm_type