第十三节 huggingface的trainner解读与Demo
文章目录
- 前言
- 一、trainer和TrainingArguments训练与预测完整Demo
- 1、数据构建
- 2、TrainingArguments构建
- 3、Trainer初始化
- 4、模型训练
- 5、模型推理
- 6、完整demo代码
- 7、完整运行结果
- 二、辅助函数
- 1、yield返回内容
- 2、迭代器中断恢复迭代demo
- 3、yield from结构
- 4、torch.Generator()的状态generator.get_state()函数