fix UnboundLocalError

evaluate_data.py

@@ -55,6 +55,7 @@ def collate_fn(data):
 
 loader = torch.utils.data.DataLoader(tokenized_test, batch_size=32, collate_fn=collate_fn)
 device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
+print(device)
 
 ner_model = ner_model.eval()
 
@@ -64,7 +65,7 @@ def get_metrics_trf():
     y_true, logits = [], []
 
     for input_ids, token_type_ids, attention_mask, labels in tqdm(loader):
-        ner_model = ner_model.to(device)
+        ner_model.to(device)
         with torch.no_grad():
             logits.extend(
                 ner_model(