使用 Unsloth 更快地训练 LLM – 在 24 小时而不是 30 天内训练自定义 ChatGPT AI

训练大型语言模型是 AI 开发的关键部分,但这也是一个可能需要花费大量时间并消耗大量计算能力的过程。这就是 Moonshot 的 Unsloth 的用武之地,可让您更快地训练 LLM。他们创建了一个新的软件包,在训练这些模型的速度和效率方面取得了重大进步。它旨在与 NVIDIA、Intel 和 AMD 等顶级公司的各种图形处理单元 (GPU) 配合使用。

Unsloth AI 训练的特点

  • 速度提高 30 倍。羊驼需要 3 小时而不是 85 小时。
  • 内存使用量减少 60%,允许批处理增加 6 倍。
  • 使用我们的 Max 产品,精度损失 0% 或精度提高 +20%。
  • 无需新硬件,只需更改软件。
  • 通过我们的 Max 产品支持 NVIDIA、Intel 和 AMD GPU。
  • 手动自动分级和链式矩阵乘法优化。
  • 用 OpenAI 的 Triton 语言重写了所有内核。
  • 通过 xformers 和 Tri Dao 的实现来获得 Flash Attention。
  • 免费的开源版本使微调速度提高了 2 倍,内存减少了 50%。

想象一下,能够将 85 小时的训练时间缩短到仅 3 小时。或者在 24 小时内而不是 30 天内训练自己的 ChatGPT。这就是我们谈论的 Unsloth AI 软件的改进。这不仅仅是向前迈出的一小步;这是一个巨大的飞跃,可以使您的工作速度提高 30 倍。此外,该软件可以将内存使用量减少 60%,这意味着您可以一次运行更大批量的数据。这种优化可以让你利用你所拥有的计算资源做更多的事情。

Unsloth AI 的软件包含高级功能,有助于其出色表现。它包括 PyTorch 中的自定义 autograd 实现、数学优化以及使用 OpenAI 的 Triton 语言重新设计的内核。该软件还使用了一种称为 Flash 注意力机制的东西来进一步提高其功能。

使用 Unsloth 更快地训练 LLM 是 AI 模型

更快地训练 LLM

无论您的预算或需求是多少,Unsloth AI 都有一个适合您的计划。他们提供不同的级别,包括免费计划、Onslot Pro 和 Onslot Max。每个都为您提供不同级别的训练加速和内存使用改进。

当您将 Unsloth AI 的软件与其他框架进行比较时,例如 Hugging Face 的变形金刚中的框架,它确实很突出。当您处理大型数据集时,它特别好,它提供了令人印象深刻的速度优势,使其成为 AI 开发人员的绝佳选择。

使用 T4 GPU 或 Google Colab

该软件最好的事情之一是它旨在与许多开发人员已经拥有的硬件一起使用。例如,您可以在标准 T4 GPU 上训练模型,这意味着您可以使用 Google Colab 等平台,而无需投资昂贵的专用硬件。

由 NVIDIA 开发的 T4 GPU 是其 Tesla 系列 GPU 的一部分,该系列专为数据中心和 AI 工作负载而设计。概述其主要功能和预期用途非常重要:

  • 架构:T4基于图灵架构,该架构也用于NVIDIA的游戏和专业可视化产品中。这种架构以其效率和性能而闻名,尤其是在人工智能和机器学习任务中。
  • 人工智能和机器学习:T4 的主要应用之一是人工智能和机器学习。它支持各种 AI 框架,并为 AI 推理和训练任务提供加速。其架构针对这些操作进行了优化,使其成为 AI 工作负载繁重的环境中的热门选择。
  • Tensor Cores:T4(和图灵架构)的一个显着特征是它的 Tensor Core。这些是旨在加速深度学习任务的专用内核。它们在执行神经网络计算中常见的矩阵运算方面非常有效。
  • 能源效率:T4 以其能源效率而著称。它为其尺寸和功耗提供了大量的计算能力,使其成为优先考虑能源效率的数据中心的有吸引力的选择。
  • 多功能性:除了 AI 和 ML,T4 还用于其他数据中心工作负载,如图形渲染、视频处理和通用计算(得益于其 CUDA 内核)。
  • 外形尺寸和部署:T4 紧凑、薄型设计使其能够适应各种服务器和数据中心配置。这种灵活性对于希望在不需要专门的硬件设置的情况下集成 GPU 加速的企业来说是有益的。
  • 精度计算:T4 支持混合精度计算,允许它根据需要调整其精度水平以优化性能或精度。这在 AI 工作负载中特别有用,在这些工作负载中,神经网络训练和推理的不同阶段可以从不同的精度级别中受益。

将 Unsloth AI 的软件添加到您当前的项目中很容易。您不必对代码库进行重大更改,并且该软件在输入数据时是用户友好的。它支持 Alpaca 提示模板或格式,因此您可以轻松上手。使用 Unsloth AI 的软件完成模型训练后,您可以对其进行优化,然后使用其他软件包进行推理和部署。这创造了一个流畅的工作流程,可以为您节省资金并帮助您更快地开发 AI 项目。

Unsloth AI 的新软件包是一个强大的工具,它正在改变开发人员训练大型语言模型的方式。它显著减少了训练时间和内存要求,可与各种 GPU 配合使用,并且易于集成到您现有的项目中。使用此软件,您可以加快 AI 开发并在竞争激烈的 AI 世界中保持领先地位。Unsloth AI 正在帮助开创 AI 模型训练的新时代,成为该领域的一部分是一个激动人心的时刻。

原创文章,作者:校长,如若转载,请注明出处:https://www.yundongfang.com/Yun271655.html

(0)
打赏 微信扫一扫不于多少! 微信扫一扫不于多少! 支付宝扫一扫礼轻情意重 支付宝扫一扫礼轻情意重
上一篇 2023年12月22日 下午10:24
下一篇 2023年12月22日 下午10:24

相关推荐