Unverified Commit 35ed78c4 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix amc doc (#3004)

parent 92738382
......@@ -4,7 +4,7 @@ This example shows us how to use AMCPruner example.
## Step 1: train a model for pruning
Run following command to train a mobilenetv2 model:
```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:
```
......
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