-
Notifications
You must be signed in to change notification settings - Fork 515
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
docstring for FNOGNO #202
docstring for FNOGNO #202
Conversation
Thanks @dhpitt, this is great! Can you just use numpy style docstrings, see e.g. https://github.com/neuraloperator/neuraloperator/blob/main/neuralop/models/fno.py. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @dhpitt, left some comments.
neuralop/models/fnogno.py
Outdated
---------- | ||
in_channels : _type_ | ||
number of input channels | ||
out_channels : _type_ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
out_channels : _type_ | |
out_channels : int |
neuralop/models/fnogno.py
Outdated
|
||
Parameters | ||
---------- | ||
in_channels : _type_ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
in_channels : _type_ | |
in_channels : int |
neuralop/models/fnogno.py
Outdated
number of hidden channels in embedding block of FNO. Defaults to 256. | ||
gno_coord_dim : int, optional | ||
dimension of GNO input data. Defaults to 3. | ||
gno_coord_embed_dim : _type_, optional |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
missing type
neuralop/models/fnogno.py
Outdated
options: 'linear_kernelonly' | ||
'linear' | ||
'nonlinear_kernelonly' | ||
'nonlinear' |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Would be good to have a small explanation of each value here
neuralop/models/fnogno.py
Outdated
How to perform domain padding. Defaults to 'one-sided'. | ||
fno_fft_norm : str, optional | ||
normalization param of torch.fft to use in FNO. Defaults to 'forward' | ||
fno_SpectralConv : nn.Module, optional |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I like to have directly default is ... instead of optional since having a default value makes it optional anyway.
Thanks @dhpitt! Merging. |
FNOGNO needs a docstring! More documentation PRs to come.