metnet-3 / model /__init__.py
yzt15806542928's picture
Upload folder using huggingface_hub
38f9214 verified
Raw
History Blame Contribute Delete
253 Bytes
"""Standalone compact MetNet-3 adaptation."""
from .metnet3 import MetNet3, MetNet3Config
from .metnet3_losses import multitask_loss
from .metnet3_schema import validate_batch
__all__ = ["MetNet3", "MetNet3Config", "multitask_loss", "validate_batch"]