AI 炼丹系列(Python): 深度学习从练气到化神,是否结丹看你的(2)

本文介绍了如何使用HuggingFace的datasets库加载和处理GPT2XL模型的训练数据,包括本地jsonl格式的数据管理和远程数据下载。作者展示了如何创建自定义数据类和配置,以及如何通过datasetsAPI进行数据集的分割和查看。
摘要由CSDN通过智能技术生成
问题:

1. 深度学习训练的常规步骤
2. 微调模型的几个思考方向 → 把次序调后以便连贯
3. 构建自己的数据集用于微调大模型
4. 各种微调方式的实现

编写代码:


模型训练的数据保存格式有很多种,datasets库提供了相关的方法从网上下载并加载导入,所以就不需要重复造轮子了。datasets是HuggingFace开发的,所以默认了从其网站下载数据的功能。在这里我们实现本地加载数据功能。

我们使用jsonl格式,以方便编辑和浏览。它和json的区别就是,每行都是一个json对象(也可以理解为python中的dict对象)。按咱的惯例先定义路径和数据格式:


cur_path = os.getcwd()

class DataPathEnum(str, Enum):
    GPT2XL_TRAIN_DATA_DIR = "gpt2xl"

    def __str__(self):
        return os.path.join(cur_path, "data", self.value)


数据文件路径和文件名为:

question, novel, konwledge是数据分类的名称,代表我要训练问答、小说、知识之类的内容。后面带_e的文件代表的是验证数据文件,两种文件里的格式一致。
question.jsonl其中的数据格式如下:(你完全可以自定义)


                
  • 15
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值