Skip to content

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.

4 participants