File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
examples/model_compression/pp-minilm Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -223,7 +223,7 @@ export FT_MODELS=../finetuning/ppminilm-6l-768h/models/CLUEWSC2020/1e-4_32
223
223
224
224
sh prune.sh CLUEWSC2020 5e-5 16 50 128 0 ${FT_MODELS} 0.75
225
225
```
226
- 其中每个参数依次表示:CLUE 中的任务名称、学习率、batch size、epoch 数、最大序列长度、gpu id、学生模型的地址、裁剪后宽度比例列表。执行完成后,模型保存的路径位于 ` pruned_models/CLUEWSC2020/0.75/ ` 。
226
+ 其中每个参数依次表示:CLUE 中的任务名称、学习率、batch size、epoch 数、最大序列长度、gpu id、学生模型的地址、裁剪后宽度比例列表。执行完成后,模型保存的路径位于 ` pruned_models/CLUEWSC2020/0.75/best_model/ ` 。
227
227
228
228
<a name =" 裁剪后模型精度 " ></a >
229
229
Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ python -u ./prune.py --model_type ernie \
29
29
--num_train_epochs ${PRE_EPOCHS} \
30
30
--logging_steps 100 \
31
31
--save_steps 100 \
32
- --output_dir ./pruned_models/$TASK_NAME /0.75/ \
32
+ --output_dir ./pruned_models/$TASK_NAME /0.75/best_model \
33
33
--device gpu \
34
34
--width_mult_list ${WIDTH_LIST}
35
35
You can’t perform that action at this time.
0 commit comments