fix loss normalization for ddp training

haoneng.lhn

2023-12-08 ace42e5043be992c3a324fc637ba0594f10ad3b6

1
2
3
4
5
6
7
8
9
one    1
two    2
three    3
four    4
five    5
six    6
seven    7
eight    8
nine    9