Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -273,7 +273,7 @@ def hybrid_score(cross_encoder_score, semantic_score, weight_cross=0.7, weight_s
 #     # Return sorted indices
 #     return [idx for idx, _ in indexed_scores]
 
-def cross_encoder_reranking(query_text, doc_texts, model, tokenizer, batch_size=8, max_length=2048):
+def cross_encoder_reranking(query_text, doc_texts, model, tokenizer, batch_size=64, max_length=2048):
     device = next(model.parameters()).device
     cross_scores = []
     query_emb = embed_text_list([query_text])[0]