File size: 318 Bytes
404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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"]
|