thak123 commited on
Commit
55edb3b
1 Parent(s): 20d5db8

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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
- _, o2 = self.bert(
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