File size: 218 Bytes
03f6091
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
# -*- coding: utf-8 -*-
from .bert import BERTEncoder
from .xlmr import XLMREncoder
from .encoder_base import Encoder

str2encoder = {"BERT": BERTEncoder, "XLMR": XLMREncoder}

__all__ = ["BERTEncoder", "XLMREncoder"]