Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -1,19 +1,19 @@
 import os
 import json
+import torch
 import argparse
 import numpy as np
-import torch
-import torch.nn.functional as F
 from tqdm import tqdm
 from torch import Tensor
-from sentence_transformers import SentenceTransformer
+import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModel
 from sklearn.cluster import AgglomerativeClustering
+from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
-# embedder = SentenceTransformer("all-MiniLM-L6-v2").to(device)
+# embedder = SentenceTransformer("all-MiniLM-L12-v2").to(device)
 embedder = SentenceTransformer("AI-Growth-Lab/PatentSBERTa").to(device)