searcher

Searcher interface for sparsity algorithms.

Classes

BaseSparseSearcher

A generic sparse mask searching algorithm.

class BaseSparseSearcher

Bases: BaseSearcher

A generic sparse mask searching algorithm.

property default_search_config: Dict[str, Any]

Get the default config for the searcher.

property default_state_dict: Dict[str, Any]

Return default state dict.

Search for sparse mask.

sanitize_search_config(config)

Sanitize the search config dict.

Parameters:

config (Dict[str, Any] | None) –

Return type:

Dict[str, Any]