Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "2f5272c78abb6d61b1e881b14c99ba4233ecb161"
Unverified Commit 35ed78c4 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix amc doc (#3004)

parent 92738382
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/model_compress/amc/README.md
View file @ 35ed78c4
...@@ -4,7 +4,7 @@ This example shows us how to use AMCPruner example. ...@@ -4,7 +4,7 @@ This example shows us how to use AMCPruner example.
## Step 1: train a model for pruning ## Step 1: train a model for pruning
Run following command to train a mobilenetv2 model: Run following command to train a mobilenetv2 model:
```bash ```bash
python3 amc_train.py --model_type mobilenetv2 --n_epochs 50 python3 amc_train.py --model_type mobilenetv2 --n_epoch 50
``` ```
Once finished, saved checkpoint file can be found at: Once finished, saved checkpoint file can be found at:
``` ```
......
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