Update model.py
Browse files
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(
|