Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -169,7 +169,7 @@ def get_detailed_instruct(task_description: str, query: str) -> str:
     """Create an instruction-formatted query"""
     return f'Instruct: {task_description}\nQuery: {query}'
 
-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):
     """
     Rerank document texts based on query text using cross-encoder model