neural_compressor.common.utils.save_load

Module Contents

Functions

save_config_mapping(config_mapping, qconfig_file_path)	Save config mapping to json file.
load_config_mapping(qconfig_file_path, config_name_mapping)	Reload config mapping from json file.

neural_compressor.common.utils.save_load.save_config_mapping(config_mapping, qconfig_file_path)

Save config mapping to json file.

Parameters:

• config_mapping (dict) – config mapping.

• qconfig_file_path (str) – path to saved json file.

neural_compressor.common.utils.save_load.load_config_mapping(qconfig_file_path, config_name_mapping)

Reload config mapping from json file.

Parameters:

• qconfig_file_path (str) – path to saved json file.

• config_name_mapping (dict) – map config name to config object. For example: ConfigRegistry.get_all_configs()[“torch”]

Returns:

config mapping.

Return type:

config_mapping (dict)