Chapter02 model doesn't converge

[sfeng0909]

Hi, I ran the code in main.py in chapter 2 and found that the loss explodes when training the model while the loss seems decrease in book. Is there any problem with the model settings?

[rajat-packt]

Hey @GuanhuaWang, would you be able to help out our reader with this?