neural_compressor.common.utils.utility

Module Contents

Classes

LazyImport	Lazy import python module till use.
CpuInfo	CPU info collection.

Functions

singleton(cls)	Singleton decorator.
dump_elapsed_time([customized_msg])	Get the elapsed time for decorated functions.
set_random_seed(seed)	Set the random seed in config.
set_workspace(workspace)	Set the workspace in config.
set_resume_from(resume_from)	Set the resume_from in config.
set_tensorboard(tensorboard)	Set the tensorboard in config.
log_quant_execution(func)

neural_compressor.common.utils.utility.singleton(cls)

Singleton decorator.

class neural_compressor.common.utils.utility.LazyImport(module_name)

Lazy import python module till use.

class neural_compressor.common.utils.utility.CpuInfo

CPU info collection.

neural_compressor.common.utils.utility.dump_elapsed_time(customized_msg='')

Get the elapsed time for decorated functions.

Parameters:

customized_msg (string, optional) – The parameter passed to decorator. Defaults to None.

neural_compressor.common.utils.utility.set_random_seed(seed: int)

Set the random seed in config.

neural_compressor.common.utils.utility.set_workspace(workspace: str)

Set the workspace in config.

neural_compressor.common.utils.utility.set_resume_from(resume_from: str)

Set the resume_from in config.

neural_compressor.common.utils.utility.set_tensorboard(tensorboard: bool)

Set the tensorboard in config.