"docs/en_US/vscode:/vscode.git/clone" did not exist on "b3a085d85b20f5325928ed9b5962f46bdaedda10"
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
trialConcurrency: 1
trialGpuNumber: 0
tuner:
name: grid
name: GridSearch
The full example can be found :githublink:`here <examples/model_compress/pruning/config.yml>`
......
......@@ -373,6 +373,6 @@ if __name__ == '__main__':
print(params)
args.sparsity = params['sparsity']
args.pruner = params['pruner']
args.model = params['pruner']
args.model = params['model']
main(args)
......@@ -15,4 +15,4 @@ trialCommand: python3 basic_pruners_torch.py --nni
trialConcurrency: 1
trialGpuNumber: 0
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