File size: 177 Bytes
cb18693
 
 
 
 
1
2
3
4
5
6
"""Model package for the CNO reproduction."""

from .FNO import CNO2d, build_model, count_trainable_parameters

__all__ = ["CNO2d", "build_model", "count_trainable_parameters"]