Back to Sentence Transformers

Trainer

docs/package_reference/sentence_transformer/trainer.md

5.4.1538 B
Original Source

Trainer

SentenceTransformerTrainer

{eval-rst}
.. autoclass:: sentence_transformers.sentence_transformer.trainer.SentenceTransformerTrainer
    :members:
    :inherited-members:
    :exclude-members: autocast_smart_context_manager, collect_features, compute_loss_context_manager, evaluation_loop, floating_point_ops, get_decay_parameter_names, get_optimizer_cls_and_kwargs, init_hf_repo, log_metrics, metrics_format, num_examples, num_tokens, predict, prediction_loop, prediction_step, save_metrics, save_state, training_step