some problems about the expermental result

[galaxysan]

I have follow your code about the loss function，but my result is very bad, can you ensure your
code about this part is right? because i find some questions about this.

[niluthpol]

I have tested the code and it is working fine. Please share your training log. Then, I will be able to help you more.