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.options
• neural_compressor.utils.pytorch
• neural_compressor.utils.utility

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 warining 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.
set_random_seed(seed)	Set the random seed in config.

Attributes

OPTIONS