aihwkit.simulator.digital_low_precision.config_utils module

Defines configuration parameters and conversions to dict structures for the quantized module base classes

class aihwkit.simulator.digital_low_precision.config_utils.CrossEntropyEstimatorParams(range_opt_method=OptMethod.golden_section, num_candidates=100, range_margin=0.5)[source]

Bases: MSEEstimatorParams

Parameters for the estimator RangeEstimators.cross_entropy. Alias of MSEEstimatorParams

Parameters:

range_opt_method (OptMethod)
num_candidates (int)
range_margin (float)

class aihwkit.simulator.digital_low_precision.config_utils.CurrentMinMaxEstimatorParams(percentile=None)[source]

Bases: object

Parameters for the estimator RangeEstimators.current_minmax

Parameters:: percentile (float | None)

percentile: float | None = None

class aihwkit.simulator.digital_low_precision.config_utils.MSEEstimatorParams(range_opt_method=OptMethod.golden_section, num_candidates=100, range_margin=0.5)[source]

Bases: object

Parameters for the estimator RangeEstimators.MSE

Parameters:

range_opt_method (OptMethod)
num_candidates (int)
range_margin (float)

num_candidates: int = 100

range_margin: float = 0.5

range_opt_method: OptMethod = 2

class aihwkit.simulator.digital_low_precision.config_utils.RunningMinMaxEstimatorParams(momentum=0.9)[source]

Bases: object

Parameters for the estimator RangeEstimators.running_minmax

Parameters:: momentum (float)

momentum: float = 0.9

aihwkit.simulator.digital_low_precision.config_utils.convert_act_config_to_kwargs_dict(act_quant_config)[source]

Converts the ActivationQuantConfig structure to a kwargs dict for the QuantizedModule base class

Parameters:: act_quant_config (ActivationQuantConfig)
Return type:: Dict[str, Any]

aihwkit.simulator.digital_low_precision.config_utils.convert_configs_to_kwargs_dict(quant_config)[source]

Converts the QuantizationConfig structure to a kwargs dict for the QuantizedModule base class

Parameters:: quant_config (QuantizationConfig)
Return type:: Dict[str, Any]

aihwkit.simulator.digital_low_precision.config_utils.convert_weight_config_to_kwargs_dict(weight_quant_config)[source]

Converts the WeightQuantConfig structure to a kwargs dict for the QuantizedModule base class

Parameters:: weight_quant_config (WeightQuantConfig)
Return type:: Dict[str, Any]