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(
|