Lisennlp / CMedGPT

基于GPT2-Chines + 医疗数据继续预训练 + 对话数据Finetune

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CMedGpt

Description

中文医疗GPT

train

bash scripts/run.sh train

eval

bash scripts/run.sh eval

如果你想使用fp16训练

安装apex

git clone https://github.com/NVIDIA/apex cd apex python setup.py install

apex问题

如果提示: if cached_x.grad_fn.next_functions[1][0].variable is not x: # error line IndexError: tuple index out of range 则请看: NVIDIA/apex#1282

finetuning

cd others

修改代码中相应的数据和模型路径

python medbert.py

About

基于GPT2-Chines + 医疗数据继续预训练 + 对话数据Finetune

License:MIT License


Languages

Language:Python 70.2%Language:Jupyter Notebook 29.0%Language:Shell 0.9%