neural_compressor.ux.utils.workload.pruning

Configuration pruning module.

Module Contents

Classes

SGDOptimizer	Configuration SGDOptimizer class.
AdamWOptimizer	Configuration AdamWOptimizer class.
AdamOptimizer	Configuration AdamOptimizer class.
Optimizer	Configuration Optimizer class.
CrossEntropyLossCriterion	Configuration CrossEntropyLossCriterion class.
SparseCategoricalCrossentropyCriterion	Configuration SparseCategoricalCrossentropyCriterion class.
KnowledgeDistillationLossCriterion	Configuration KnowledgeDistillationLossCriterion class.
IntermediateLayersKnowledgeDistillationLoss	Configuration IntermediateLayersKnowledgeDistillationLoss class.
SelfKnowledgeDistillationLoss	Configuration SelfKnowledgeDistillationLoss class.
Criterion	Configuration Criterion class.
Train	Configuration Train class.
WeightCompressionApproach	Configuration WeightCompressionApproach class.
Approach	Configuration Approach class.
Pruning	Configuration Pruning class.

Functions

parse_dict_value_to_float(→ Optional[float])	Parse value to float or None if value is None.
parse_dict_value_to_int(→ Optional[int])	Parse value to float or None if value is None.

class neural_compressor.ux.utils.workload.pruning.SGDOptimizer(data: Dict[str, Any] = {})

Configuration SGDOptimizer class.

class neural_compressor.ux.utils.workload.pruning.AdamWOptimizer(data: Dict[str, Any] = {})

Configuration AdamWOptimizer class.

class neural_compressor.ux.utils.workload.pruning.AdamOptimizer(data: Dict[str, Any] = {})

Configuration AdamOptimizer class.

class neural_compressor.ux.utils.workload.pruning.Optimizer(data: Dict[str, Any] = {})

Configuration Optimizer class.

class neural_compressor.ux.utils.workload.pruning.CrossEntropyLossCriterion(data: Dict[str, Any] = {})

Configuration CrossEntropyLossCriterion class.

class neural_compressor.ux.utils.workload.pruning.SparseCategoricalCrossentropyCriterion(data: Dict[str, Any] = {})

Configuration SparseCategoricalCrossentropyCriterion class.

class neural_compressor.ux.utils.workload.pruning.KnowledgeDistillationLossCriterion(data: Dict[str, Any] = {})

Configuration KnowledgeDistillationLossCriterion class.

class neural_compressor.ux.utils.workload.pruning.IntermediateLayersKnowledgeDistillationLoss(data: Dict[str, Any] = {})

Configuration IntermediateLayersKnowledgeDistillationLoss class.

class neural_compressor.ux.utils.workload.pruning.SelfKnowledgeDistillationLoss(data: Dict[str, Any] = {})

Configuration SelfKnowledgeDistillationLoss class.

class neural_compressor.ux.utils.workload.pruning.Criterion(data: Dict[str, Any] = {})

Configuration Criterion class.

class neural_compressor.ux.utils.workload.pruning.Train(data: Dict[str, Any] = {})

Configuration Train class.

class neural_compressor.ux.utils.workload.pruning.WeightCompressionApproach(data: Dict[str, Any] = {})

Configuration WeightCompressionApproach class.

class neural_compressor.ux.utils.workload.pruning.Approach(data: Dict[str, Any] = {})

Configuration Approach class.

class neural_compressor.ux.utils.workload.pruning.Pruning(data: Dict[str, Any] = {})

Configuration Pruning class.

neural_compressor.ux.utils.workload.pruning.parse_dict_value_to_float(data: dict, key: str) → float | None

Parse value to float or None if value is None.

neural_compressor.ux.utils.workload.pruning.parse_dict_value_to_int(data: dict, key: str) → int | None

Parse value to float or None if value is None.