27 lines
755 B
Python
27 lines
755 B
Python
"""Baseline model definitions."""
|
|
|
|
from airfrans_frontier.models.coordinate_encoding import CoordinateEncoder, CoordinateEncodingSpec
|
|
from airfrans_frontier.models.film import FourierFiLMMLP
|
|
from airfrans_frontier.models.frontier import (
|
|
DeepONetBranchTrunk,
|
|
LocalPointTransformer,
|
|
NeRFCFDMultiRes,
|
|
PointContextPerceiver,
|
|
RasterFNOUNet,
|
|
SirenConditionedINR,
|
|
)
|
|
from airfrans_frontier.models.mlp import EncodedPointwiseMLP, PointwiseMLP
|
|
|
|
__all__ = [
|
|
"CoordinateEncoder",
|
|
"CoordinateEncodingSpec",
|
|
"DeepONetBranchTrunk",
|
|
"EncodedPointwiseMLP",
|
|
"FourierFiLMMLP",
|
|
"LocalPointTransformer",
|
|
"NeRFCFDMultiRes",
|
|
"PointContextPerceiver",
|
|
"PointwiseMLP",
|
|
"RasterFNOUNet",
|
|
"SirenConditionedINR",
|
|
]
|