mirror of https://github.com/PyRetri/PyRetri.git
16 lines
334 B
Python
16 lines
334 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from yacs.config import CfgNode
|
||
|
|
|
||
|
|
# from .registry import BACKBONES
|
||
|
|
from .backbone.backbone_base import BACKBONES
|
||
|
|
|
||
|
|
|
||
|
|
def get_model_cfg() -> CfgNode:
|
||
|
|
cfg = CfgNode()
|
||
|
|
for name in BACKBONES:
|
||
|
|
cfg[name] = CfgNode()
|
||
|
|
cfg[name]["load_checkpoint"] = ""
|
||
|
|
cfg["name"] = "unknown"
|
||
|
|
return cfg
|