crystina-z
commited on
Commit
•
98fc9bd
1
Parent(s):
ef8b55b
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
Tevatron reranker
|
2 |
+
|
3 |
+
```
|
4 |
+
examples/reranker/reranker_train.py \
|
5 |
+
--output_dir reranker_xlmr.bs-32.epoch-1.mmarco \
|
6 |
+
--model_name_or_path xlm-roberta-large --save_steps 20000 --dataset_name crystina-z/mmarco-train:all --fp16 --per_device_train_batch_size 4 --gradient_accumulation_steps 8 --train_n_passages 8 --learning_rate 5e-6 --q_max_len 16 --p_max_len 128 --num_train_epochs 1 --logging_steps 500 --dataloader_num_workers 4 --overwrite_output_dir
|
7 |
+
```
|