self.block_type = block_type
    self.layers = layers
    self.num_classes = num_classes
    self.input_channels = input_channels
    self.cardinality = cardinality
    self.base_width = base_width
    self.stem_width = stem_width
    self.stem_type = stem_type
    self.avg_down = avg_down
    super().__init__(**kwargs)