FNO2D outputs constant prediction value about geo-fno HOT 4 CLOSED

It sounds like there is a bug in the setup, so the model can't learn anything.
Just to check, when you call the FNO model, did you use out = model(u, x_in= x_in, x_out= x_out), where x_in is (x_in, y_in) and x_out is (x_out, y_out).

And to begin, it may be helpful to define the deformation IPHI as a constant function (just return x in the forward function).

A few more questions:
what is your Number of samples?
what are the training and testing/validation errors?

from geo-fno.

Hi! Thanks for the prompt reply!

I did use out = model(u, x_in = x_in, x_out = x_out). u have shape [batch_size, Number of mesh points, 3].
Just to clarify, my x_out and y_out are x coordinates and y coordinates of the output field respectively.

Number of samples: 3207
Both training and validation errors floats around a constant after just 2 or 3 epochs.

Will definitely try out having the deformation IPHI as a constant function.

Thanks!

from geo-fno.

Hi! With the deformation IPHI set as a constant function, the output I am receiving is the same as before, which is a constant value for all points on the mesh.

from geo-fno.

Hi, thank you for your time, as I have found the issue lies within the output data construction.

Best,
Jin

from geo-fno.