Skip to content

Commit 1757a4f

Browse files
authored
fix distill readme, and upadate aistudio link of day12 course in tutorial (#605)
1 parent 2e9de52 commit 1757a4f

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

education/day12.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Day12 预训练模型小型化与部署实战作业辅导
22

33
本教程旨在辅导同学如何完成 AI Studio课程——[『NLP打卡营』实践课12:预训练模型小型化与部署实战
4-
](https://aistudio.baidu.com/aistudio/projectdetail/1920541)课后作业。
4+
](https://aistudio.baidu.com/aistudio/projectdetail/2017520)课后作业。
55

66
## 1. 对ERNIE-Gram进行fine-tuning得到教师模型
77
由于我们的蒸馏是在中文情感分析ChnSentiCorp任务上,因此我们需要对PaddleNLP提供的ERNIE-Gram在我们的任务上进行Fine-tuning。下面是详细的步骤:

examples/model_compression/distill_lstm/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ python -u ./run_glue.py \
6565
--logging_steps 10 \
6666
--save_steps 10 \
6767
--output_dir ../model_compression/distill_lstm/pretrained_models/$TASK_NAME/ \
68-
--n_gpu 1 \
68+
--device gpu \
6969

7070
```
7171

examples/model_compression/tinybert/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ python -u ./run_glue.py \
5454
--logging_steps 1 \
5555
--save_steps 500 \
5656
--output_dir ./tmp/$TASK_NAME/ \
57-
--n_gpu 1 \
57+
--device gpu \
5858

5959
```
6060

0 commit comments

Comments
 (0)