def METHOD_NAME(self):
    config = super().METHOD_NAME()
    config.update(
        {
            "include_rescaling": self.include_rescaling,
            # Remove batch dimension from `input_shape`
            "input_shape": self.input_shape[1:],
            "stackwise_channels": self.stackwise_channels,
            "stackwise_depth": self.stackwise_depth,
            "activation": self.activation,
        }
    )
    return config