Source code for xmodaler.modeling.decode_strategy.build

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

DECODE_STRATEGY_REGISTRY = Registry("DECODE_STRATEGY")
DECODE_STRATEGY_REGISTRY.__doc__ = """
Registry for decode strategy
"""

[docs]def build_beam_searcher(cfg):
    beam_search = DECODE_STRATEGY_REGISTRY.get(cfg.DECODE_STRATEGY.NAME)(cfg)
    return beam_search

[docs]def build_greedy_decoder(cfg):
    greedy_decoder = DECODE_STRATEGY_REGISTRY.get("GreedyDecoder")(cfg)
    return greedy_decoder