training loss problem

[zhangzy003]

Thanks for your sharing your code with me!
When I try to run your train.py, some error occurs.I found that the training loss is too much, and it leads to the bad result.
I am looking forward to your reply!