convergence comparison NN vs WNN

[qideng7]

Hi,
First of all, thank you for this great work. I rerun your NN and WNN but couldn't achieve the claimed result. Here are screenshots of NN and WNN, respectively:


WNN hasn't converged and MSE loss was higher than NN.
It will be great to have your comments on that result.
Thank you!