Available Model Architectures¤
If you would like your architecture added, please submit a pull request.
Standard¤
Conditioned¤
Note
Currently only scalar parameter conditioning is available.
Architecture | Model Name | Description |
---|---|---|
FNO | ResNet with FourierBasicBlock |
Addition based conditioning in both spatial and spectral domain. |
U-Net-modern | Unet |
Addition and AdaGN style conditioning in the spatial domain. |
UF-Net | FourierUnet |
Addition and AdaGN style conditioning in the spatial domain, Addition in the spectral domain. |
Model Architecture Registry Philosophy¤
While in principle we can make every architecture fully configurable via configuration files, we find it can affect the readability of the code quite a bit. Feel free to open issues or pull-requests for further configuration ability or any other suggestions for managing the configurability-readability tradeoffs.