neural_compressor.utils

Utils: provide useful methods and auxiliary functionalities.

Submodules

• neural_compressor.utils.collect_layer_histogram
• neural_compressor.utils.constant
• neural_compressor.utils.create_obj_from_config
• neural_compressor.utils.kl_divergence
• neural_compressor.utils.load_huggingface
• neural_compressor.utils.logger
• neural_compressor.utils.neural_insights_utils
• neural_compressor.utils.options
• neural_compressor.utils.pytorch
• neural_compressor.utils.utility
• neural_compressor.utils.weights_details

Package Contents

Classes

LayerHistogramCollector

The collector of the histogram by layer.

Functions

log(level, msg, *args, **kwargs)	Output log with the level as a parameter.
info(msg, *args, **kwargs)	Output log with the info level.
debug(msg, *args, **kwargs)	Output log with the debug level.
warn(msg, *args, **kwargs)	Output log with the warning level.
warning(msg, *args, **kwargs)	Output log with the warning level (Alias of the method warn).
error(msg, *args, **kwargs)	Output log with the error level.
fatal(msg, *args, **kwargs)	Output log with the fatal level.
alias_param(param_name, param_alias)	Decorator for aliasing a param in a function.

Attributes

OPTIONS