about the code's feasibility

[LeslieChen233]

Is it the real code used to implement your experiments? I have found the loss in your code is not consistent with the description in your paper and there are many matrix dimension errors in your batch training phase. e.g.