Skip to content

Commit cd03820

Browse files
committed
update readme
1 parent 5b902f3 commit cd03820

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

examples/finetune/embedder/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,9 +75,11 @@ cd FlagEmbedding/scripts
7575
python add_reranker_score.py \
7676
--input_file toy_finetune_data_minedHN.jsonl \
7777
--output_file toy_finetune_data_score.jsonl \
78-
--range_for_sampling 2-200 \
79-
--negative_number 15 \
80-
--use_gpu_for_searching
78+
--reranker_name_or_path BAAI/bge-reranker-v2-m3 \
79+
--devices cuda:0 cuda:1 \
80+
--cache_dir ./cache/model \
81+
--reranker_query_max_length 512 \
82+
--reranker_max_length 1024
8183
```
8284

8385
- **`input_file`**: path to save JSON data with mined hard negatives for finetuning

0 commit comments

Comments
 (0)