FIx aux loss scale with calculate_per_token_loss. - #3105

FIx aux loss scale with calculate_per_token_loss.#3105

Victarry wants to merge 1 commit into

NVIDIA:devNVIDIA/Megatron-LM:devfrom

Victarry:denliu/fix_per_token_aux_lossVictarry/Megatron-LM:denliu/fix_per_token_aux_lossCopy head branch name to clipboard