跳到正文
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