File size: 269 Bytes
d4cbafd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from .mtr_decoder import MTRDecoder
__all__ = {
'MTRDecoder': MTRDecoder,
}
def build_decoder(config, use_pre_norm, **kwargs):
model = __all__[config.NAME](
config=config,
use_pre_norm=use_pre_norm,
**kwargs
)
return model
|