From 7b77db0772997adaf04f492f46525c2973dd495b Mon Sep 17 00:00:00 2001
From: lijianshe02 <48898730+lijianshe02@users.noreply.github.com>
Date: Sun, 26 Apr 2020 17:55:58 +0800
Subject: [PATCH] update prune demo README.md file test=develop (#243)

---
 demo/prune/README.md | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/demo/prune/README.md b/demo/prune/README.md
index 138a315b..c31f1791 100755
--- a/demo/prune/README.md
+++ b/demo/prune/README.md
@@ -39,10 +39,14 @@ export CUDA_VISIBLE_DEVICES=0
 python train.py \
 --model "MobileNet" \
 --pruned_ratio 0.31 \
---data "mnist"
+--data "mnist"  \
+--criterion "l1_norm"
 ```
 
 其中，`model`用于指定待裁剪的模型。`pruned_ratio`用于指定各个卷积层通道数被裁剪的比例。`data`选项用于指定使用的数据集。
+`criterion` 选项用于指定所使用的剪裁算法策略，现在支持`l1_norm`, `bn_scale`, `geometry_median`。默认为`l1_norm`。可以
+设置该参数以改变剪裁算法策略。该目录下的四个shell脚本文件是在ResNet34, MobileNetV1, MobileNetV2等三个模型上进行的四组
+`criterion`设置为`geometry_median`的实验，可以直接运行脚本文件启动剪裁实验。
 
 执行`python train.py --help`查看更多选项。
 
-- 
GitLab