vis4d/zoo/base/__init__.py

"""Model Zoo base."""

from .callable import get_callable_cfg
from .dataloader import get_inference_dataloaders_cfg, get_train_dataloader_cfg
from .optimizer import get_lr_scheduler_cfg, get_optimizer_cfg
from .pl_trainer import get_default_pl_trainer_cfg
from .runtime import get_default_callbacks_cfg, get_default_cfg

__all__ = [
    "get_callable_cfg",
    "get_train_dataloader_cfg",
    "get_inference_dataloaders_cfg",
    "get_optimizer_cfg",
    "get_lr_scheduler_cfg",
    "get_default_cfg",
    "get_default_callbacks_cfg",
    "get_default_pl_trainer_cfg",
]