searcher

Searcher interface for sparsity algorithms.

Classes

BaseSparseSearcher

A generic sparse mask searching algorithm.

class BaseSparseSearcher

Bases: BaseSearcher

A generic sparse mask searching algorithm.

config: dict[str, Any]

constraints: dict[str, str | float | dict | None]

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.

deployment: dict[str, str] | None

dummy_input: Any | tuple

forward_loop: Callable[[Module], None] | None

model: Module

run_search(): Search for sparse mask.

sanitize_search_config(config)

Sanitize the search config dict.

Parameters:: config (dict[str, Any] | None)
Return type:: dict[str, Any]