Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

modify the config bug about Automatic Model Pruning using NNI Tuners #3541

Merged
merged 3 commits into from
Apr 28, 2021

Conversation

BUPTLYH
Copy link
Contributor

@BUPTLYH BUPTLYH commented Apr 16, 2021

The bug is about config error, and the specific modification location is as follows:
1. Modify the code in "nni/examples/model_compress/pruning/basic_pruners_torch.py" , 378 line :args.model = params['model']
2. Modify the code in "nni/examples/model_compress/pruning/config.yml", 18 line :name: GridSearch

@ghost
Copy link

ghost commented Apr 16, 2021

CLA assistant check
All CLA requirements met.

@SparkSnail SparkSnail requested review from J-shang and colorjam April 19, 2021 02:55
@@ -15,4 +15,4 @@ trialCommand: python3 basic_pruners_torch.py --nni
trialConcurrency: 1
trialGpuNumber: 0
tuner:
name: grid
name: GridSearch
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you~ By the way, you can also modify the English document docs/en_US/Compression/AutoPruningUsingTuners.rst line 67.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok

@@ -64,7 +64,7 @@ op_type 'Conv2d' 表示在 PyTorch 框架下定义在 :githublink:`default_layer
trialConcurrency: 1
trialGpuNumber: 0
tuner:
name: grid
name: GridSearch
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The Chinese doc is auto-translated from the English doc, directly modify will meet an error, please revert it.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

OK

@JunweiSUN JunweiSUN mentioned this pull request Apr 20, 2021
@QuanluZhang QuanluZhang merged commit 3725c8f into microsoft:master Apr 28, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants