File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -94,6 +94,7 @@ python normal_regular_prune.py --percent 0.5 --model models_save/nin_preprune.pt
94
94
```
95
95
python normal_regular_prune.py --percent 0.5 --normal_regular 8 --model models_save/nin_preprune.pth --save models_save/nin_prune.pth
96
96
```
97
+ 或
97
98
```
98
99
python normal_regular_prune.py --percent 0.5 --normal_regular 16 --model models_save/nin_preprune.pth --save models_save/nin_prune.pth
99
100
```
@@ -142,9 +143,14 @@ python main.py --W 2 --A 2 --refine ../prune/models_save/nin_gc_refine.pth
142
143
```
143
144
cd WbWtAb/bn_merge
144
145
```
146
+ --W 权重W量化取值(据训练时W量化(FP/三值/二值)情况而定)
145
147
#### 融合并保存融合前后model
146
148
```
147
- python bn_merge.py
149
+ python bn_merge.py --W 2
150
+ ```
151
+ 或
152
+ ```
153
+ python bn_merge.py --W 3
148
154
```
149
155
#### 融合前后model对比测试
150
156
```
You can’t perform that action at this time.
0 commit comments