File tree Expand file tree Collapse file tree 3 files changed +295
-8
lines changed Expand file tree Collapse file tree 3 files changed +295
-8
lines changed Original file line number Diff line number Diff line change 86
86
from .reformer .tokenizer import *
87
87
from .mobilebert .modeling import *
88
88
from .mobilebert .tokenizer import *
89
+ from .chinesebert .modeling import *
90
+ from .chinesebert .tokenizer import *
Original file line number Diff line number Diff line change 42
42
from paddlenlp .transformers import PretrainedModel , register_base_model
43
43
from paddlenlp .transformers .bert .modeling import BertPooler , BertPretrainingHeads
44
44
45
- # from .fusion_embedding import FusionBertEmbeddings
46
-
47
45
__all__ = [
48
46
"ChineseBertModel" ,
49
47
"ChineseBertPretrainedModel" ,
55
53
]
56
54
57
55
58
- # fusion_embedding.py
59
56
class PinyinEmbedding (nn .Layer ):
60
57
def __init__ (self ,
61
58
pinyin_map_len : int ,
@@ -65,6 +62,7 @@ def __init__(self,
65
62
Pinyin Embedding Layer.
66
63
67
64
Args:
65
+ pinyin_map_len (int): the size of pinyin map, which about 26 Romanian characters and 6 numbers.
68
66
embedding_size (int): the size of each embedding vector.
69
67
pinyin_out_dim (int): kernel number of conv.
70
68
You can’t perform that action at this time.
0 commit comments