Dario Coscia
603f56d264
Neural Operator fix and addition
...
* Building FNO for 1D/2D/3D data
* Fixing bug in trunk/branch net in DeepONEt
* Fixing type check bug in spectral conv
* Adding tests for FNO
* Fixing bug in Fourier Layer (conv1d/2d/3d)
2023-11-17 09:51:29 +01:00
Dario Coscia
83ecdb0eab
Fourier Block and minor fixes
...
* Adding fourier block 1d/2d/3d
* Adding docs to SpectralConvBlock1D/2D/3D and to FourierBlock1D/2D/3D
* Adding tests for fourier block
2023-11-17 09:51:29 +01:00
Dario Coscia
b7f9694cb9
Spectral Convolution Addition
...
* Implementing 1D/2D/3D spectral conv
* Implementing tests for 1D/2D/3d spectral conv
2023-11-17 09:51:29 +01:00
Dario Coscia
15ecaacb7c
Layers and Models update PR
...
* add residual block
* add test conv and residual block
* modify FFN kwargs
2023-11-17 09:51:29 +01:00
