PulseAugur
实时 11:40:39
English(EN) Unsloth vs Axolotl vs TRL: 87% of Your Fine-Tuning VRAM Goes to a Tensor You Never Wrote

已识别微调 VRAM 瓶颈:损失张量消耗大部分内存

一项技术分析显示,在 LoRA 微调过程中,很大一部分 VRAM 被临时的交叉熵损失张量消耗,而不是模型本身。该张量仅短暂存在以产生单个标量输出,在 GPT OSS 20B 等模型中,它占序列长度相关内存使用的 95.2%。对 UnslothAxolotlTRL 等微调框架的比较突显了它们在如何管理这个内存密集型张量方面存在主要差异,从而影响了整体 VRAM 效率。 AI

影响 突出了当前 LLM 微调方法中关键的 VRAM 效率低下问题,表明围绕损失张量管理的框架优化可以实现显著的内存节省。

排序理由 对 AI 模型微调过程的技术分析。[lever_c_demoted from research: ic=1 ai=1.0]

在 Towards AI 阅读 →

AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →

已识别微调 VRAM 瓶颈:损失张量消耗大部分内存

报道来源 [1]

  1. Towards AI TIER_1 English(EN) · Chew Loong Nian - AI ENGINEER ·

    Unsloth vs Axolotl vs TRL: 87% of Your Fine-Tuning VRAM Goes to a Tensor You Never Wrote

    <div class="medium-feed-item"><p class="medium-feed-image"><a href="https://pub.towardsai.net/unsloth-vs-axolotl-vs-trl-87-of-your-fine-tuning-vram-goes-to-a-tensor-you-never-wrote-d21b8326d89d?source=rss----98111c9905da---4"><img src="https://cdn-images-1.medium.com/max/1400/1*W…