config

Default configurations for sparsity modes.

Classes

`ExportSparseConfig`	Configuration (empty!) for the `"export_sparse"` mode.
`SparseGPTConfig`	Configuration for the `"sparse_gpt"` mode.
`SparseMagnitudeConfig`	Configuration for the `"sparse_magnitude"` mode.

class ExportSparseConfig

Bases: ModeloptBaseConfig

Configuration (empty!) for the "export_sparse" mode.

model_config = {'extra': 'forbid', 'validate_assignment': True}: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class SparseGPTConfig

Bases: ModeloptBaseRuleConfig

Configuration for the "sparse_gpt" mode.

model_config = {'extra': 'allow', 'validate_assignment': True}: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

nn_conv2d: SparseConv2dConfig | None | dict[str, SparseConv2dConfig | None]

nn_linear: SparseLinearConfig | None | dict[str, SparseLinearConfig | None]

class SparseMagnitudeConfig

Bases: ModeloptBaseRuleConfig

Configuration for the "sparse_magnitude" mode.

model_config = {'extra': 'allow', 'validate_assignment': True}: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

nn_conv2d: SparseConv2dConfig | None | dict[str, SparseConv2dConfig | None]

nn_linear: SparseLinearConfig | None | dict[str, SparseLinearConfig | None]