Reshard Collective로 강화학습 가중치 리프릿 가속하기
Accelerating Reinforcement Learning Weight Refit with the Reshard Collective
강화학습(RL) 사후 학습에서는 현재 정책 모델로 추론을 실행해 학습 궤적을 실시간으로 생성한다. 학습 효율을 위해 추론과 학습은 분리된 GPU 자원에서 병렬로 실행되는 경우가 많다. 정책 모델은 학습 GPU에서만 업데이트되지만, 시스템은 추론 쪽의 가중치를 주기적으로 갱신해야 한다. 기사는 이 가중치 리프릿(refit) 과정을 Reshard Collective로 가속하는 방법을 다룬다.
분리된 학습·추론 구성에서 주기적 가중치 갱신 비용이 Reshard Collective로 줄어들 수 있음을 NVIDIA 기술 블로그가 설명한다.
원문 출처
NVIDIA Technical Blog