Update vocab.py

This commit is contained in:
leo 2019-12-04 17:21:55 +08:00
parent 7c5be2240f
commit 392fc28528
1 changed files with 5 additions and 0 deletions

View File

@ -12,6 +12,9 @@ SPECIAL_TOKENS_KEYS = [
"sep_token",
"bos_token",
"eos_token",
"head_token",
"tail_token",
]
SPECIAL_TOKENS_VALUES = [
@ -22,6 +25,8 @@ SPECIAL_TOKENS_VALUES = [
"[SEP]",
"[BOS]",
"[EOS]",
"HEAD",
"TAIL",
]
SPECIAL_TOKENS = OrderedDict(zip(SPECIAL_TOKENS_KEYS, SPECIAL_TOKENS_VALUES))