How to use the saved model？

[leanty]

How to load a model saved as JAX in the models folder to predict values？Any suggestions are welcome.

[SimLab120]

use orbax. I use it to checkpoint and then predict the model

[benmoseley]

Also have a look at fbpinns.analysis: https://github.com/benmoseley/FBPINNs/blob/main/fbpinns/analysis.py
We have a function for loading saved models there.