Update DistilBERT.py

DistilBERT.py

@@ -137,6 +137,9 @@ def train(epoch):
             break
         loss.backward()
         optimizer.step()
+def loss_fn(outputs, targets):
+    return torch.nn.BCEWithLogitsLoss()(outputs, targets)
+optimizer = torch.optim.Adam(params =  model_DB.parameters(), lr=LEARNING_RATE)
 
-for epoch in range(3):
-    train(epoch)
+loaded_model_path = './model_DB_1.pt'
+model_DB.load_state_dict(torch.load(loaded_model_path, map_location=torch.device('cpu')))