crystina-z commited on
Commit
98fc9bd
1 Parent(s): ef8b55b

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
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
+ ```