Skip to content

Commit 76056cf

Browse files
committed
Merge branch 'master' of https://github.com/UKPLab/sentence-transformers into v4.0-release
2 parents 3fa7983 + 095de27 commit 76056cf

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

docs/cross_encoder/training_overview.md

+14-8
Original file line numberDiff line numberDiff line change
@@ -620,11 +620,14 @@ The :class:`~sentence_transformers.cross_encoder.trainer.CrossEncoderTrainer` is
620620
621621
from datasets import load_dataset
622622
623-
from sentence_transformers.cross_encoder import CrossEncoder, CrossEncoderModelCardData
623+
from sentence_transformers.cross_encoder import (
624+
CrossEncoder,
625+
CrossEncoderModelCardData,
626+
CrossEncoderTrainer,
627+
CrossEncoderTrainingArguments,
628+
)
624629
from sentence_transformers.cross_encoder.evaluation import CrossEncoderNanoBEIREvaluator
625630
from sentence_transformers.cross_encoder.losses import CachedMultipleNegativesRankingLoss
626-
from sentence_transformers.cross_encoder.trainer import CrossEncoderTrainer
627-
from sentence_transformers.cross_encoder.training_args import CrossEncoderTrainingArguments
628631
629632
# Set the log level to INFO to get more information
630633
logging.basicConfig(format="%(asctime)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S", level=logging.INFO)
@@ -759,15 +762,18 @@ The :class:`~sentence_transformers.cross_encoder.trainer.CrossEncoderTrainer` is
759762
from datasets import load_dataset
760763
761764
from sentence_transformers import SentenceTransformer
762-
from sentence_transformers.cross_encoder import CrossEncoder, CrossEncoderModelCardData
765+
from sentence_transformers.cross_encoder import (
766+
CrossEncoder,
767+
CrossEncoderModelCardData,
768+
CrossEncoderTrainer,
769+
CrossEncoderTrainingArguments,
770+
)
763771
from sentence_transformers.cross_encoder.evaluation import (
764772
CrossEncoderNanoBEIREvaluator,
765773
CrossEncoderRerankingEvaluator,
766774
)
767-
from sentence_transformers.cross_encoder.losses.BinaryCrossEntropyLoss import BinaryCrossEntropyLoss
768-
from sentence_transformers.cross_encoder.trainer import CrossEncoderTrainer
769-
from sentence_transformers.cross_encoder.training_args import CrossEncoderTrainingArguments
770-
from sentence_transformers.evaluation.SequentialEvaluator import SequentialEvaluator
775+
from sentence_transformers.cross_encoder.losses import BinaryCrossEntropyLoss
776+
from sentence_transformers.evaluation import SequentialEvaluator
771777
from sentence_transformers.util import mine_hard_negatives
772778
773779
# Set the log level to INFO to get more information

0 commit comments

Comments
 (0)