Sentence Transformers v5.7.0:修复梯度错误,训练提速3.9倍

tomaarsen · x · 2026-08-06

Sentence Transformers 发布 v5.7.0 版本,重点修复了 GradCache 机制中的隐蔽 Bug(如重排序模型在反向传播中重放不同的 dropout 掩码导致梯度偏差),并将所有缓存损失重构为共享超类。

性能方面,新增基于 token 数量的 mini-batch 打包功能,在 Natural Questions 数据集上结合 Flash Attention,训练速度提升 3.9 倍且不损失质量。此外,修复了 model.compile() 在推理时静默失效的问题,开启 CUDA graphs 后在 bge-small 模型上实测推理速度提升约 3 倍。

所属事件:Sentence Transformers v5.7.0 发布修复梯度Bug并大幅提速(3 条相关)→

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →