File size: 571 Bytes
9b33fca
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"""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",
]