from sentence_transformers import SentenceTransformerTrainer trainer = SentenceTransformerTrainer( model=model, args=train_args, train_dataset=dataset["train"], eval_dataset=dataset["valid"], loss=loss, evaluator=[evaluator_recall_train, evaluator_recall_valid], ) trainer.train()