Skip to content

fix(loss): count num_valid_samples from the sample mask, not the batch dim - #3850

Open
tianyi-zhang-02 wants to merge 2 commits into
NVIDIA-NeMo:mainfrom
tianyi-zhang-02:fix-num-valid-samples
Open

fix(loss): count num_valid_samples from the sample mask, not the batch dim#3850
tianyi-zhang-02 wants to merge 2 commits into
NVIDIA-NeMo:mainfrom
tianyi-zhang-02:fix-num-valid-samples