Helmholtz pytorch predicted problem

[FemLearner]

I try the tensorflow version of the Helmholtz problem and it was working correctly:

But when I try the version of pytorch , and whatever i did(changing the layers or iter) ,the result was bad , if there is some parameters make this problem happen?I checked out the code and I think there is nothing wrong?

[aho38]

I fixed it if you wanna check out mine (https://github.com/aho38/Pytorch-PINN-Marine-Lake/tree/main/Data_Discovery). The issue was in autograd of the parameters. It's computing the wrong input.