十一、搭建自己的GPT模型

本文介绍了如何使用nanoGPT搭建并训练一个可以续写古诗词的模型,包括环境搭建、数据准备、训练过程以及可能出现的错误处理。通过训练唐诗数据集,生成了ckpt.pt模型文件,可用于古诗词的生成。

    nanoGPT,是用于培训/微调中型GPT的最简单、最快的存储库。因为代码非常简单,所以很容易满足需求,从头开始训练新模型,或者微调。基于GPT-2 1.3B模型,优点是cpu也可以跑,简单,快速。LLaMa的模型训练太耗费gpu,很多人也跑不了,所以暂时选择这个。
    项目地址:https://github.com/karpathy/nanoGPT
1、搭建环境
    git clone https://github.com/karpathy/nanoGPT
    pip install transformers tiktoken tqdm
2、快速体验demo模型
    在莎士比亚的作品中训练一个角色级的GPT,默认250轮会保存一次模型;在训练差不多可以直接ctrl+c停掉。
    如果仔细观察,会发现我们正在训练一个GPT,其上下文大小最多为256个字符,384个功能通道,它是一个6层的Transformer,每层有6个头。
    模型保存路径:D:\Simon\nanoGPT\out-shakespeare-char\ckpt.pt
    # 下载数据集
    $ python data/shakespeare_char/prepare.py
    # 使用GPU训练,在一个A100 GPU上,此训练运行大约需要3分钟。没有gpu,使用参数--dev

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

vandh

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值