Skip to content

Conversation

yiliu30
Copy link
Contributor

@yiliu30 yiliu30 commented Jan 23, 2024

Type of Change

Feature, implement the #1445 (comment)
API changed or not: YES

Description

  • API change
+ get_all_config_set()
- TuningConfig(quant_configs=...)
+ TuningConfig(config_set=get_all_config_set(), max_trials=4)
  • Usage
from neural_compressor.torch import TuningConfig, autotune, get_all_config_set
tune_config = TuningConfig(config_set=get_all_config_set(), max_trials=4)
  • FWK owner TODO
class AlgoConfig(BaseConfig):
    ...
    # fwk owner needs to implement the `get_config_set_for_tuning` for each algo config.
    @classmethod
    def get_config_set_for_tuning(cls):
        ...

Test

Pre-CI

Dependency Change?

None

@yiliu30 yiliu30 marked this pull request as ready for review January 23, 2024 03:33
@yiliu30 yiliu30 changed the title Add default tuning space Add default config set for tuning Jan 24, 2024
@yiliu30 yiliu30 merged commit fa8e66a into master Jan 25, 2024
@yiliu30 yiliu30 deleted the ly/dt branch January 25, 2024 09:23
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.

5 participants