Skip to content

Add tolerable_loss to TuningConfig #1579

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 27 commits into from
Jan 30, 2024
Merged

Add tolerable_loss to TuningConfig #1579

merged 27 commits into from
Jan 30, 2024

Conversation

Kaihui-intel
Copy link
Contributor

Type of Change

feature

Description

Usage:

from neural_compressor.torch import RTNConfig, TuningConfig
custom_tune_config = TuningConfig(config_set=[RTNConfig(weight_bits=[4, 6, 8])],  tolerable_loss=-0.01)

#1445 (comment)

Expected Behavior & Potential Risk

the expected behavior that triggered by this PR

How has this PR been tested?

3x UT

Dependency Change?

None

@Kaihui-intel Kaihui-intel requested a review from yiliu30 January 26, 2024 07:35
@yiliu30 yiliu30 changed the title Enhance the TuningConfig by adding tolerable_loss Add tolerable_loss to TuningConfig Jan 26, 2024
Signed-off-by: Kaihui-intel <[email protected]>
Signed-off-by: Kaihui-intel <[email protected]>
@Kaihui-intel Kaihui-intel requested a review from yiliu30 January 30, 2024 02:35
Signed-off-by: Kaihui-intel <[email protected]>
@chensuyue chensuyue merged commit fb61428 into master Jan 30, 2024
@chensuyue chensuyue deleted the kaihui/acc_loss branch January 30, 2024 09:47
mengniwang95 pushed a commit that referenced this pull request Jan 31, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants