Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -223,6 +223,7 @@ def cross_encoder_reranking(query_text, doc_texts, model, tokenizer, batch_size=
     return [idx for idx, _ in indexed_scores]
 
 def main():
+    base_directory = os.getcwd()
     parser = argparse.ArgumentParser(description='Re-rank patents using cross-encoder scoring (training queries only)')
     parser.add_argument('--pre_ranking', type=str, default='shuffled_pre_ranking.json',
                         help='Path to pre-ranking JSON file')
@@ -248,7 +249,7 @@ def main():
     parser.add_argument('--device', type=str, default='cuda' if torch.cuda.is_available() else 'cpu',
                         help='Device to use (cuda/cpu)')
     parser.add_argument('--base_dir', type=str, 
-                        default='datasets',
+                        default=f'{base_directory}/datasets',
                         help='Base directory for data files')
     
     args = parser.parse_args()