Skip to content

Commit

Permalink
fix loss name in documentation of CachedMultipleNegativesRankingLoss (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
JINO-ROHIT authored Jan 23, 2025
1 parent b765d85 commit c9d3c96
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,7 @@ def __init__(
"anchor": ["It's nice weather outside today.", "He drove to work."],
"positive": ["It's so sunny.", "He took the car to the office."],
})
loss = losses.CachedGISTEmbedLoss(model, mini_batch_size=64)
loss = losses.CachedMultipleNegativesRankingLoss(model, mini_batch_size=64)
trainer = SentenceTransformerTrainer(
model=model,
Expand Down

0 comments on commit c9d3c96

Please sign in to comment.