File size: 313 Bytes
d4cbafd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | from .mtr_encoder import MTREncoder
from .eth_encoder import ETHEncoder
__all__ = {
'MTREncoder': MTREncoder,
'ETHEncoder': ETHEncoder,
}
def build_context_encoder(config, use_pre_norm):
model = __all__[config.NAME](
config=config,
use_pre_norm=use_pre_norm
)
return model
|