Running the Module multiple times should be perfectly fine.
At this point, I would recommend using torchviz 18.
You can use it to plot the graph associated with one or more Tensors.
In particular here, you can make sure that everything is connected as you expect to the right place.
Note that for readability if your pinn model is large, you can replace it with a simpler version with less layers.