Skip to content

Commit

Permalink
multi-GPU support for mine_hard_negatives
Browse files Browse the repository at this point in the history
  • Loading branch information
alperctnkaya authored and tomaarsen committed Oct 17, 2024
1 parent 6ad366c commit ab85d71
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sentence_transformers/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -533,7 +533,7 @@ def mine_hard_negatives(
faiss_batch_size: int = 16384,
use_faiss: bool = False,
verbose: bool = True,
use_multiple_gpus = True,
use_multiple_gpus = False,
) -> Dataset:
"""
Add hard negatives to a dataset of (anchor, positive) pairs to create (anchor, positive, negative) triplets or
Expand Down

0 comments on commit ab85d71

Please sign in to comment.