Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
phlobo authored Nov 27, 2023
1 parent 1c22491 commit da46eb1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,7 @@ prediction = rr.rerank_batch(candidates['test'], ce_dataset['test'])

#### Note on Memory Usage
In most examples and benchmarks, we use 64 candidates as a batch size for the cross-encoder, which usually fit into 48GB of GPU memory.
If you encounter memory issue, you can try reducing this number and/or using a smaller BERT model. See: [Issue #22](https://github.com/hpi-dhc/xmen/issues/22#issuecomment-1827648900)
If you encounter memory issues, you can try reducing this number and/or using a smaller BERT model. See: [Issue #22](https://github.com/hpi-dhc/xmen/issues/22#issuecomment-1827648900)

### Pre-trained Cross-encoders

Expand Down

0 comments on commit da46eb1

Please sign in to comment.