Skip to content

Conversation

yiliu30
Copy link
Contributor

@yiliu30 yiliu30 commented Feb 1, 2024

Type of Change

feature
API changed or not: add API to get the default set

Description

from neural_compressor.torch.quantization import TuningConfig, autotune, get_rtn_double_quant_config_set
def eval_acc_fn(model) -> float:
    ...

custom_tune_config = TuningConfig(config_set=get_rtn_double_quant_config_set())
q_model = autotune(model=build_simple_torch_model(), tune_config=custom_tune_config, eval_fns=eval_acc_fn)

How has this PR been tested?

Pre-CI

Dependency Change?

None

@yiliu30 yiliu30 added Auto-Tune PyTorch Related to PyTorch F/W labels Feb 1, 2024
@yiliu30 yiliu30 merged commit b8d98eb into master Feb 2, 2024
@yiliu30 yiliu30 deleted the ly/db_tuning branch February 2, 2024 03:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Auto-Tune PyTorch Related to PyTorch F/W
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants