monoXLMR.pft-mmarco / README.md
crystina-z's picture
Create README.md
98fc9bd
|
raw
history blame contribute delete
No virus
454 Bytes
Tevatron reranker
```
examples/reranker/reranker_train.py \
--output_dir reranker_xlmr.bs-32.epoch-1.mmarco \
--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
```