Update README.md

README.md

@@ -13,11 +13,11 @@ mobilenetv2:
 
 详细模型结构见官方论文：
 EfficientNet
-https://arxiv.org/pdf/1905.11946.pdf
+* https://arxiv.org/pdf/1905.11946.pdf
 mobilenetv2
-https://arxiv.org/pdf/1801.04381.pdf
+* https://arxiv.org/pdf/1801.04381.pdf
 mobilenetv3
-https://arxiv.org/pdf/1905.02244.pdf
+* https://arxiv.org/pdf/1905.02244.pdf
 ## 算法原理
 CNN原理
 ## 数据集
@@ -32,8 +32,8 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk
 cd mobilenet_prune
 pip3 install -r requirements.txt
 ```
-
-## 剪枝
+## 训练
+### 剪枝
 efficientnet
 ```
 python3 main.py --mode prune --model efficientnet --batch-size 128 --restore ./checkpoints/efficientnet.pt --dataset cifar10  --method l1 --speed-up 2.11 --global-pruning
@@ -46,7 +46,7 @@ mobilenetv3
 ```
 python3 main.py --mode prune --model mobilenetv3 --batch-size 128 --restore ./checkpoints/mobilenetv3.pt --dataset cifar10  --method l1 --speed-up 2.11 --global-pruning
 ```
-## 测试
+### 测试
 efficientnet
 ```
 python3 main.py --mode test --model efficientnet --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-efficientnet/cifar10_efficientnet_l1.pt --dataset cifar10  --method l1 --speed-up 2.11 --global-pruning
@@ -60,6 +60,7 @@ mobilenetv3
 python3 main.py --mode test --model mobilenetv3 --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-mobilenetv3/cifar10_mobilenetv3_l1.pt --dataset cifar10  --method l1 --speed-up 2.11 --global-pruning
 ```
 ## result
+
 ### 精度
 method l1
 | model | Base Acc | Pruned Acc |