thank you ptrblck, I noticed this post 42 in Github and I thought I should use
reduction = 'sum'

If I understand right, you mean that if I leave the reduction = 'mean', it will be fine and I won’t worry about the loss value and the learning rate.
Thank you, it helps me a lot.