fix the loss backward issue

Commit

6 years ago

fix the loss backward issue (cherry picked from commit 566468cc984c6ec7e10dfc62b5b4191781a99cd2)