SentenceTransformerModel.train_model