Source code for xmodaler.modeling.decoder.build

# Copyright 2021 JD.com, Inc., JD AI
"""
@author: Yehao Li
@contact: yehaoli.sysu@gmail.com
"""
from xmodaler.utils.registry import Registry

DECODER_REGISTRY = Registry("DECODER")
DECODER_REGISTRY.__doc__ = """
Registry for decoder
"""

[docs]def build_decoder(cfg):
    decoder = DECODER_REGISTRY.get(cfg.MODEL.DECODER)(cfg) if len(cfg.MODEL.DECODER) > 0 else None
    return decoder

[docs]def add_decoder_config(cfg, tmp_cfg):
    if len(tmp_cfg.MODEL.DECODER) > 0:
        DECODER_REGISTRY.get(tmp_cfg.MODEL.DECODER).add_config(cfg)