* Adding Residual MLP * Adding test Residual MLP * Modified relative import Continuous Conv
17 lines
484 B
Python
17 lines
484 B
Python
__all__ = [
|
|
'ContinuousConvBlock',
|
|
'ResidualBlock',
|
|
'EnhancedLinear',
|
|
'SpectralConvBlock1D',
|
|
'SpectralConvBlock2D',
|
|
'SpectralConvBlock3D',
|
|
'FourierBlock1D',
|
|
'FourierBlock2D',
|
|
'FourierBlock3D',
|
|
]
|
|
|
|
from .convolution_2d import ContinuousConvBlock
|
|
from .residual import ResidualBlock, EnhancedLinear
|
|
from .spectral import SpectralConvBlock1D, SpectralConvBlock2D, SpectralConvBlock3D
|
|
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
|