Update evaluation.py

evaluation.py

@@ -1,7 +1,6 @@
 import evaluate
 import torch
 from enum import Enum
-from scripts.gputils import print_gpu_utilization, clear_gpu_mem
 from tqdm import tqdm
 
 
@@ -28,7 +27,6 @@ def classify_assertions_in_sentences(sentences, model, tokenizer, batch_size=32)
             outputs = model(**batch)
         predicted_labels = torch.argmax(outputs.logits, dim=1)
         predictions.append(predicted_labels)
-        print_gpu_utilization()
     return torch.cat(predictions)