ITMO-NSS-team / torch_DE_solver

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs
https://torch-de-solver.readthedocs.io
BSD 3-Clause "New" or "Revised" License
59 stars 9 forks source link

Fourier layers and casuality #30

Closed aminevdam closed 1 year ago

aminevdam commented 1 year ago

1) module models.py is added, it consists of classes for new neural network's architecture. 2) small refactoring for solution and solver classes. (it will make it easier to install adaptive lambdas.) 3) new method casual_loss for O/PDE depending on time variable. (example_Allen_Cahn.py, example_LV_hunter_prey.py, example_wave_deepxde.py)