ldy/bge-finetune
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Revised for training

Browse Source
This commit is contained in:
ldy
2025-07-23 14:54:46 +08:00
parent 6dbd2f3281
commit 229f6bb027
32 changed files with 59884 additions and 11081 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/run_validation_suite.py
View File

@@ -265,7 +265,8 @@ class ValidationSuiteRunner:
try:
cmd = [
'python', 'scripts/compare_retriever.py',
'python', 'scripts/compare_models.py',
'--model_type', 'retriever',
'--finetuned_model_path', self.config['retriever_model'],
'--baseline_model_path', self.config.get('retriever_baseline', 'BAAI/bge-m3'),
'--data_path', dataset_path,
@@ -301,7 +302,8 @@ class ValidationSuiteRunner:
try:
cmd = [
'python', 'scripts/compare_reranker.py',
'python', 'scripts/compare_models.py',
'--model_type', 'reranker',
'--finetuned_model_path', self.config['reranker_model'],
'--baseline_model_path', self.config.get('reranker_baseline', 'BAAI/bge-reranker-base'),
'--data_path', dataset_path,