SensitivityAnalysisSparsifier.Config

Component: SensitivityAnalysisSparsifier

class SensitivityAnalysisSparsifier.Config

Bases: Sparsifier.Config

All Attributes (including base classes)

pre_train_model_path: str = ''

analyzed_sparsity: float = 0.8

max_analysis_batches: int = 0

max_skipped_weight: int = 0

pre_analysis_path: str = ''

sparsity: float = 0.8

iterative_pruning: bool = True

pruning_iterations: int = 2

start_sparsity_ratio: float = 0.5

Default JSON

{
    "pre_train_model_path": "",
    "analyzed_sparsity": 0.8,
    "max_analysis_batches": 0,
    "max_skipped_weight": 0,
    "pre_analysis_path": "",
    "sparsity": 0.8,
    "iterative_pruning": true,
    "pruning_iterations": 2,
    "start_sparsity_ratio": 0.5
}