Skip to content

Latest commit

 

History

History
10 lines (9 loc) · 430 Bytes

README.md

File metadata and controls

10 lines (9 loc) · 430 Bytes

gpt2

  • finetune_gpt2.py 加载预训练模型并微调
  • train_raw_data.txt 微调GPT2使用的训练数据抽样
  • test_raw_data.txt 微调GPT2使用的测试数据抽样

Bert

  • finetune_bert.py 加载预训练模型并微调分类任务
  • finetune_test_data_sample.txt 类目预测任务采样训练数据
  • finetune_train_data_sample.txt 类目预测任务采样验证数据
  • cate_dict.txt 类目id映射数据