753e275
1
2
3
4
5
6
7
8
9
10
11
12
13
14
_MODEL_DICT = {} def register_model(name): def decorator(cls): _MODEL_DICT[name] = cls return cls return decorator def get_model(cfg): return _MODEL_DICT[cfg.type](cfg)
