A suggestion about the loss function in the paper.

[liuhao-lh]

Maybe the square should be outside the brackets in the second term of the loss Eq.(2). 🤔