can't reduce the value of “MainLoss ”

[zhk779]

Hi，thanks for your code ,your job is amazing.
I meeting a problem when I'm training the model. The value of “MainLoss ” is shaking just in the beginning，can you give me some advice？
I only modify the value of "batch_size" to 2, because I only have one gpu.
Thanks.