Update model.py
Browse files
model.py
CHANGED
@@ -17,7 +17,7 @@ class BERTBaseUncased(nn.Module):
|
|
17 |
print("Model")
|
18 |
|
19 |
def forward(self, ids, mask, token_type_ids):
|
20 |
-
|
21 |
ids,
|
22 |
attention_mask=mask,
|
23 |
token_type_ids=token_type_ids
|
|
|
17 |
print("Model")
|
18 |
|
19 |
def forward(self, ids, mask, token_type_ids):
|
20 |
+
output = self.bert(
|
21 |
ids,
|
22 |
attention_mask=mask,
|
23 |
token_type_ids=token_type_ids
|