facebookresearch / GENRE

Autoregressive Entity Retrieval
Other
765 stars 103 forks source link

Details on training GENRE for page-level document retrieval #69

Closed amy-hyunji closed 2 years ago

amy-hyunji commented 2 years ago

Hi, I would like to train GENRE from BART on page-level document retrieval with only DPR data (GENRE only DPR data). I noticed that in the appendix, the training was done on 128 GPUs. However, when I try training on 8 32GB V100 GPUs with scripts_genre/train.sh, it only uses about 10.3GB of the 32GB. I wonder if I missed something. Also, will there be a difference in the batch size if I reduce the number of GPUs? Thanks :)