Unverified Commit 3725c8fc authored by BUPTLYH's avatar BUPTLYH Committed by GitHub
Browse files

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

parent 063d6b74
...@@ -64,7 +64,7 @@ Then, define a ``config`` file in YAML to automatically tuning model, pruning al ...@@ -64,7 +64,7 @@ Then, define a ``config`` file in YAML to automatically tuning model, pruning al
trialConcurrency: 1 trialConcurrency: 1
trialGpuNumber: 0 trialGpuNumber: 0
tuner: tuner:
name: grid name: GridSearch
The full example can be found :githublink:`here <examples/model_compress/pruning/config.yml>` The full example can be found :githublink:`here <examples/model_compress/pruning/config.yml>`
......
...@@ -373,6 +373,6 @@ if __name__ == '__main__': ...@@ -373,6 +373,6 @@ if __name__ == '__main__':
print(params) print(params)
args.sparsity = params['sparsity'] args.sparsity = params['sparsity']
args.pruner = params['pruner'] args.pruner = params['pruner']
args.model = params['pruner'] args.model = params['model']
main(args) main(args)
...@@ -15,4 +15,4 @@ trialCommand: python3 basic_pruners_torch.py --nni ...@@ -15,4 +15,4 @@ trialCommand: python3 basic_pruners_torch.py --nni
trialConcurrency: 1 trialConcurrency: 1
trialGpuNumber: 0 trialGpuNumber: 0
tuner: tuner:
name: grid name: GridSearch
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment