hugging_face_tutorials
hugging_face_tutorials copied to clipboard
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映射数据