Skip to content

Backend Pytorch: Support msffn and stmsffn#2047

Open
tianyeeT wants to merge 3 commits intolululxvi:masterfrom
tianyeeT:ffn
Open

Backend Pytorch: Support msffn and stmsffn#2047
tianyeeT wants to merge 3 commits intolululxvi:masterfrom
tianyeeT:ffn

Conversation

@tianyeeT
Copy link
Copy Markdown
Contributor

All changes are tested in my local machine. All good.

@lululxvi
Copy link
Copy Markdown
Owner

Format the code using black https://github.com/psf/black

@echen5503
Copy link
Copy Markdown
Contributor

@tianyeeT are you still working on this?

@echen5503
Copy link
Copy Markdown
Contributor

Can you provide the testing code that you used to test on local machine?

@tianyeeT
Copy link
Copy Markdown
Contributor Author

tianyeeT commented Mar 2, 2026

@echen5503 I tested the scripts examples/pinn_forward/wave_1d.py and examples/pinn_forward/Poisson_multiscale_1d.py. You just need to specify the backend as PyTorch.

@echen5503
Copy link
Copy Markdown
Contributor

Testing Results: Both scripts reach low loss on mixed-frequency problems, consistent with paper results.
Docs: LGTM
Code: Faithful translation from paddle version (same structure and functionality).

@lululxvi I recommend merge

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants