十一、搭建自己的GPT模型

    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,使用参数--device=cpu
    $ python train.py config/train_shakespeare_char.py
    $ python train.py config/train_shakespeare_char.py --device=cpu --compile=False --eval_iters&

  • 0
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

vandh

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

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

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

打赏作者

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

抵扣说明:

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

余额充值