NVIDIA-NeMo / NVIDIA-NeMo/RL

Refactor source manager for policy, reference and reward model

Open
#1,100 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

# Description

For now, the resource for policy is computed from the whole resource, inference resource and rewaed model resource. This leads to the inability to customize the training resources.
For example, in non-colocated condition, if there are 2 nodes and 8 gpus per nodes, 4 gpus for inference, 4 gpus for rm env. Then the resource for policy can be 8 gpus on 1 node, and also can be 2 nodes and each with 4 gpus.

# Refactor

Modify config, and specified resource for each actor. Set up a separate resource management module.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.