thak123 commited on
Commit
5ed46c0
·
1 Parent(s): 2635faf

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -14,6 +14,7 @@ class BERTBaseUncased(nn.Module):
14
  # self.out = nn.Linear(256, 3)
15
 
16
  nn.init.xavier_uniform_(self.out.weight)
 
17
 
18
  def forward(self, ids, mask, token_type_ids):
19
  _, o2 = self.bert(
 
14
  # self.out = nn.Linear(256, 3)
15
 
16
  nn.init.xavier_uniform_(self.out.weight)
17
+ print("Model")
18
 
19
  def forward(self, ids, mask, token_type_ids):
20
  _, o2 = self.bert(