From 555fa6d8073620e4dc743d7c9b975511964d17f8 Mon Sep 17 00:00:00 2001 From: Nils Reimers Date: Wed, 23 Dec 2020 22:20:18 +0100 Subject: [PATCH] docs --- sentence_transformers/datasets/SentenceLabelDataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sentence_transformers/datasets/SentenceLabelDataset.py b/sentence_transformers/datasets/SentenceLabelDataset.py index 4b0385f1d..ae5b3f468 100644 --- a/sentence_transformers/datasets/SentenceLabelDataset.py +++ b/sentence_transformers/datasets/SentenceLabelDataset.py @@ -27,7 +27,7 @@ def __init__(self, examples: List[InputExample], samples_per_label: int = 2, wit :param examples: a list with InputExamples :param samples_per_label: - the number of consecutive, random and unique samples drawn per label + the number of consecutive, random and unique samples drawn per label. Batch size should be a multiple of samples_per_label :param with_replacement: if this is True, then each sample is drawn at most once (depending on the total number of samples per label). if this is False, then one sample can be drawn in multiple draws, but still not multiple times in the same