Skip to content

Commit d94b628

Browse files
authored
fix ppminilm scripts bug (#1490)
1 parent 55130a9 commit d94b628

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/model_compression/pp-minilm/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ export FT_MODELS=../finetuning/ppminilm-6l-768h/models/CLUEWSC2020/1e-4_32
223223

224224
sh prune.sh CLUEWSC2020 5e-5 16 50 128 0 ${FT_MODELS} 0.75
225225
```
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/`
227227

228228
<a name="裁剪后模型精度"></a>
229229

examples/model_compression/pp-minilm/pruning/prune.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ python -u ./prune.py --model_type ernie \
2929
--num_train_epochs ${PRE_EPOCHS} \
3030
--logging_steps 100 \
3131
--save_steps 100 \
32-
--output_dir ./pruned_models/$TASK_NAME/0.75/ \
32+
--output_dir ./pruned_models/$TASK_NAME/0.75/best_model \
3333
--device gpu \
3434
--width_mult_list ${WIDTH_LIST}
3535

0 commit comments

Comments
 (0)