I modified both main() and ode_system() as stated in the instructions, but I get a very high test loss. Do I have to modify other functions? I also increased the number of epochs to 10^5 as pointed in the supplementary information of the cited paper (a very interesting work I have to say).
Thank you in advance.
Additional information:
Modified lines (all in src/deeponet_pde.py):
line 104: uncommented after having commented line 102
line 231: be sure to have problem="ode"
line 260: increased epochs to 100000
I modified both
main()
andode_system()
as stated in the instructions, but I get a very high test loss. Do I have to modify other functions? I also increased the number of epochs to 10^5 as pointed in the supplementary information of the cited paper (a very interesting work I have to say). Thank you in advance.Additional information: Modified lines (all in
src/deeponet_pde.py
): line 104: uncommented after having commented line 102 line 231: be sure to haveproblem="ode"
line 260: increased epochs to 100000Example of output: