searcher — Model Optimizer 0.27.1 (original) (raw)

Searcher interface for sparsity algorithms.

Classes

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.

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_]