Đinh Trác Đức Anh commited on
Commit ·
6e42c14
1
Parent(s): a27a735
Block bmes emb option
Browse files
model.py
CHANGED
|
@@ -198,7 +198,7 @@ class MorphemeAwareRobertaModel(RobertaModel):
|
|
| 198 |
bmes_ids = bmes_tags
|
| 199 |
|
| 200 |
if inputs_embeds is None and self.block_bmes_emb == False:
|
| 201 |
-
print("Using bmes embeddings")
|
| 202 |
inputs_embeds = self.embeddings(
|
| 203 |
input_ids=input_ids,
|
| 204 |
token_type_ids=token_type_ids,
|
|
@@ -208,7 +208,7 @@ class MorphemeAwareRobertaModel(RobertaModel):
|
|
| 208 |
)
|
| 209 |
|
| 210 |
if self.block_bmes_emb == True:
|
| 211 |
-
print("Block bmes embeddings")
|
| 212 |
inputs_embeds = self.embeddings(
|
| 213 |
input_ids=input_ids,
|
| 214 |
token_type_ids=token_type_ids,
|
|
|
|
| 198 |
bmes_ids = bmes_tags
|
| 199 |
|
| 200 |
if inputs_embeds is None and self.block_bmes_emb == False:
|
| 201 |
+
# print("Using bmes embeddings")
|
| 202 |
inputs_embeds = self.embeddings(
|
| 203 |
input_ids=input_ids,
|
| 204 |
token_type_ids=token_type_ids,
|
|
|
|
| 208 |
)
|
| 209 |
|
| 210 |
if self.block_bmes_emb == True:
|
| 211 |
+
# print("Block bmes embeddings")
|
| 212 |
inputs_embeds = self.embeddings(
|
| 213 |
input_ids=input_ids,
|
| 214 |
token_type_ids=token_type_ids,
|