Skip to content

Create a FNO wrapper #22

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
pzimbrod opened this issue Feb 4, 2022 · 0 comments
Open

Create a FNO wrapper #22

pzimbrod opened this issue Feb 4, 2022 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@pzimbrod
Copy link
Contributor

pzimbrod commented Feb 4, 2022

In order to assimilate FourierLayer to the yet to implement DeepONet, it would be nice to have a Fourier Neural Operator (FNO) constructor that creates the entire architecture.

Something like:

model = FNO(in, out, latentspace, layercount, grid, modes, activation)

Instead of having to make the entire chain yourself:

layer = FourierLayer(latentspace, latentspace, grid, modes, activation)

model = Chain(
Dense(in, latentspace, activation),
layer,
layer,
[...],
Dense(latentspace, out, activation)
)

This doesn't add functionality but helps unify the API of the package. FourierLayer should still stay accessible via API as is since it offers a lot more control per layer than constructing the entire architecture with all the same global arguments.

@pzimbrod pzimbrod added the enhancement New feature or request label Feb 4, 2022
@pzimbrod pzimbrod self-assigned this Feb 4, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant