nemo_rl.telemetry.span_groups#
NeMo-RL specific span groups.
Module Contents#
Classes#
Span groups for NeMo-RL instrumentation. |
API#
- class nemo_rl.telemetry.span_groups.RLSpanGroup#
Bases:
nemo.lens.groups.SpanGroupSpan groups for NeMo-RL instrumentation.
- ROLLOUT#
‘rollout’
Rollout collection spans.
- GENERATION#
‘generation’
Text generation spans.
- LOGPROB#
‘logprob’
Log-probability computation spans.
- REWARD#
‘reward’
Reward computation spans.
- ADVANTAGE#
‘advantage’
Advantage computation spans.
- POLICY_UPDATE#
‘policy_update’
Policy gradient update spans.
- REFERENCE_POLICY#
‘reference_policy’
Reference policy log-prob computation spans.
- DATA_PROCESSING#
‘data_processing’
Data processing / batching spans.
- EFFICIENCY#
‘efficiency’
Async efficiency phases (idle / wasted accounting).
Unlike the other leaf groups these do not have one fixed bucket — the
rl.bucketcomes from the category, so emit them viainstrumentation.efficiency_span.
- ALL_GROUPS: Final[frozenset]#
None
- _PRESETS: ClassVar[dict]#
None