5 comments (5 comments)2 reactions (2 reactions)0 assignees (0 assignees)Python4,413 forks (4,413 forks)auto 404
enhancementfeature requestgood first issue
Repository metrics
- Stars
- 22,985 stars (22,985 stars)
- PR merge metrics
- Avg merge 3d 6h (Avg merge 3d 6h)125 merged PRs in 30d (125 merged PRs in 30d)
Description
Motivation
reduce_metrics seems to be used for reducing metrics across DP ranks and batches, which only simply means the values.
However, reducing metrics across DP ranks is non-trivial since some metrics like "mean" are not linear.
Contributor guide
- Research direction
- Investigate the current reduce metrics implementation in verl/utils/metric/utils.py and determine how to properly reduce non linear metrics like mean across DP ranks. Consider using atomic reductions or computing sum and count separately.
- Tech stack
- python
- Domain
- aimachine learningbackend
- Issue type
- Feature
- Prerequisites
- Python