from yacs.config import CfgNode | |
from .config.default import _CN | |
def lower_config(yacs_cfg): | |
if not isinstance(yacs_cfg, CfgNode): | |
return yacs_cfg | |
return {k.lower(): lower_config(v) for k, v in yacs_cfg.items()} | |
def get_model_cfg(): | |
cfg = lower_config(lower_config(_CN)) | |
return cfg["model"] | |