Commit Graph

8 Commits

Author SHA1 Message Date
Dario Coscia
56bd0dac7d minor updates in garom.py
* Removing `retain_graph` in backward for discriminator
* Fixing issues with different precision training for Lightining
2023-11-17 16:08:29 +01:00
Dario Coscia
c90301c204 Network handles forward for all solvers 2023-11-17 09:51:29 +01:00
Dario Coscia
8b7b61b3bd Documentation for v0.1 version (#199)
* Adding Equations, solving typos
* improve _code.rst
* the team rst and restuctore index.rst
* fixing errors

---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
2023-11-17 09:51:29 +01:00
Dario Coscia
3f9305d475 Solvers logging (#202)
* Modifying solvers to log every epoch correctly
* add `on_epoch` flag to logger
* fix bug in `pinn.py` `pts -> samples` in `_loss_phys`
* add `optimizer_zero_grad()` in garom generator training loop
* modify imports in `callbacks.py`
* fixing tests

---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
2023-11-17 09:51:29 +01:00
Nicola Demo
d654259428 add dataset and dataloader for sample points (#195)
* add dataset and dataloader for sample points
* unittests
2023-11-17 09:51:29 +01:00
Dario Coscia
2e2fe93458 GAROM solver loss update
Changing from `LpLoss` to `PowerLoss`
2023-11-17 09:51:29 +01:00
Dario Coscia
80b4b43460 GAROM loggers (#181)
* adding loggers for GAROM solver

---------

Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
2023-11-17 09:51:29 +01:00
Dario Coscia
701046661f Solvers for multiple models (#133)
* Solvers for multiple models
- Implementing the possibility to add multiple models for solvers (e.g. GAN)
- Implementing GAROM solver, see https://arxiv.org/abs/2305.15881
- Implementing tests for GAROM solver (cpu only)
- Fixing docs PINNs
- Creating a solver directory, for consistency in the package


---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-040.eduroam.sissa.it>
2023-11-17 09:51:29 +01:00