fix loss normalization for ddp training

haoneng.lhn

2023-12-08 ace42e5043be992c3a324fc637ba0594f10ad3b6

1
2
3
4
5
6
7
8
9
10
11
12
1    12
2    1
3    2
4    3
5    4
6    5
7    6
8    7
9    8
10    9
11    10
12    11