rezaakb/pinns-torch

Demo example in tutorial did not provide expected result

PepperXZC opened this issue · 2 comments

PepperXZC commented

I just directly cloned this repository, had virtual env setup and ran tutorials/0-Schrodinger.ipynb, but the resulted prediction did not reach convergence. Is there any additional setting should be done or any hyperparameters?
image

rezaakb commented

Yes, please increase the value of max_epochs in the training configuration. For instance:
trainer = pl.Trainer(accelerator='cpu', devices=1, max_epochs=60000)

  • 👍2
PepperXZC commented

Thanks for your patience!