Browse Source

Remove todo

pull/558/merge
duzx16 2 years ago
parent
commit
79e4d8ba8a
  1. 8
      ptuning/README.md

8
ptuning/README.md

@ -133,7 +133,7 @@ gradient_accumulation_steps=16
## 模型部署
将对应的demo或代码中的`THUDM/chatglm-6b`换成经过 P-Tuning 微调之后 checkpoint 的地址(在示例中为 `./output/adgen-chatglm-6b-pt-8-1e-2/checkpoint-3000`)。注意,目前的微调还不支持多轮数据,所以只有对话第一轮的回复是经过微调的。
将对应的demo或代码中的`THUDM/chatglm-6b`换成经过 P-Tuning 微调之后 checkpoint 的地址(在示例中为 `./output/adgen-chatglm-6b-pt-8-1e-2/checkpoint-3000`)。
## 使用自己的数据集
修改 `train.sh``evaluate.sh` 中的 `train_file`、`validation_file`和`test_file`为你自己的 JSON 格式数据集路径,并将 `prompt_column``response_column` 改为 JSON 文件中输入文本和输出文本对应的 KEY。
@ -189,12 +189,6 @@ gradient_accumulation_steps=16
bash train_chat.sh
```
## TODO
* [x] Support for chat data
* [x] Support for full finetuning
## 引用
```

Loading…
Cancel
Save