hahnec/torchimize

An example to use LM optimizer

sashahexe opened this issue · 2 comments

sashahexe commented

Hi,

thank you for your library, I'd like to use the LM optimizer in my PINN which is set up as a class based on pytorch. Could you share some examples of the code where you use the LM as an optimizer?

Thanks again!

hahnec commented

Hi Sasha,

Yet I only implemented a Gauss Newton optimizer, which I tested for a simple toy training example:

https://github.com/hahnec/torchimize/blob/master/tests/unit_test_raw_fit.py