Skip to content

fix(megatron): honor policy.logprob_chunk_size in the training loss path - #2872

Merged
terrykong merged 3 commits into
NVIDIA-NeMo:mainfrom
kaloyan-inherent:kally/fix-logprob-chunk-loss-path
Jun 23, 2026
Merged

fix(megatron): honor policy.logprob_chunk_size in the training loss path#2872
terrykong merged 3 commits into
NVIDIA-NeMo:mainfrom
kaloyan-inherent:kally/fix-logprob-chunk-loss-path

suffix: add fp32 for clarity

c7d05e8
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jun 22, 2026 in 1s

DCO

All commits are signed off!