LightR0 / hugging_face_tutorials

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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映射数据

About


Languages

Language:Python 100.0%