Hugging Face Blog·· 2024-03-20精选AI 评分70
GaLore:在消费级硬件上高效训练大语言模型
GaLore: Advancing Large Model Training on Consumer-grade Hardware
AI 导读
Hugging Face发布GaLore优化器,利用梯度低秩结构投影技术将优化器状态显存占用降低82.5%以上,结合8-bit量化可在RTX 4090等消费级显卡上训练7B参数模型。该方案支持动态子空间切换与层更新,已集成于transformers库,用户可通过pip安装galore-torch并使用galore_adamw等优化器进行微调。
推荐理由
GaLore通过低秩投影显著降低显存占用,配合8-bit优化器可在消费级显卡上训练大模型,提供了可复用的训练方案。
来源:Hugging Face Blog · huggingface.co