Commit
·
b6794ac
1
Parent(s):
1ae5b9a
Update DistilBERT.py
Browse files- DistilBERT.py +2 -0
DistilBERT.py
CHANGED
|
@@ -137,6 +137,8 @@ def train(epoch):
|
|
| 137 |
optimizer.step()
|
| 138 |
def loss_fn(outputs, targets):
|
| 139 |
return torch.nn.BCEWithLogitsLoss()(outputs, targets)
|
|
|
|
|
|
|
| 140 |
optimizer = torch.optim.Adam(params = model_DB.parameters(), lr=LEARNING_RATE)
|
| 141 |
|
| 142 |
loaded_model_path = './model_DB_1.pt'
|
|
|
|
| 137 |
optimizer.step()
|
| 138 |
def loss_fn(outputs, targets):
|
| 139 |
return torch.nn.BCEWithLogitsLoss()(outputs, targets)
|
| 140 |
+
|
| 141 |
+
model_DB = DistilBERTClass()
|
| 142 |
optimizer = torch.optim.Adam(params = model_DB.parameters(), lr=LEARNING_RATE)
|
| 143 |
|
| 144 |
loaded_model_path = './model_DB_1.pt'
|