I am trying to solve the 1D Bar with the help of FNO. What could be the problem and what are good ways to find it? [closed]

I have attached 2 codes above. The second code follows the structure of a fourier layer while the first code imports the second code and use it as a machine to calculate and verify correct deformation. Usually I would ha