Here the code of loss_pde = self.criterion(du_dt + u.squeeze() * du_dx, 0.01 / math.pi * du_dxx) of coefficient 0.01 / math.pi need to be (0.01 / math.pi) , and the code will be loss_pde = self.criterion(du_dt + u.squeeze() * du_dx, (0.01 / math.pi) * du_dxx).
Then the result will change. It's so amazing.
I am not aware of that the change of parentheses would result in a change in the result. Did you fix the random seed to test out whether it is due to randomness?
Here the code of
loss_pde = self.criterion(du_dt + u.squeeze() * du_dx, 0.01 / math.pi * du_dxx)
of coefficient 0.01 / math.pi need to be (0.01 / math.pi) , and the code will beloss_pde = self.criterion(du_dt + u.squeeze() * du_dx, (0.01 / math.pi) * du_dxx)
. Then the result will change. It's so amazing.