NVIDIA Developer Blog· Tanya Lenz·· 2 天前AI 评分31
NVIDIA 用 Reshard Collective 加速强化学习权重重载
Accelerating Reinforcement Learning Weight Refit with the Reshard Collective
AI 导读
NVIDIA 提出 Reshard Collective,用于加速强化学习后训练中分离式部署下的权重重载。该场景下推理与训练在独立 GPU 资源上并行运行,策略模型仅在训练 GPU 上更新,系统需定期把新权重刷新到推理侧。
来源:NVIDIA Developer Blog · developer.nvidia.com