fix loss normalization for ddp training

haoneng.lhn

2023-12-08 acb9a0fec8d8a4dabeedcbb8e08c26f66d7083f0

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
boy
boy
boy
boy
boy
boy
boy
girl
girl
pretty girl
cat
dog
dog
dog
rat
red faced clown
green caterpillar