TextEmbedder¶
- class flash.text.embedding.model.TextEmbedder(backbone='sentence-transformers/all-MiniLM-L6-v2', max_length=128, tokenizer_backbone=None, tokenizer_kwargs=None, enable_ort=False)[source]¶
The
TextEmbedder
is aTask
for generating sentence embeddings, training and validation. For more details, see embeddings.You can change the backbone to any question answering model from UKPLab/sentence-transformers using the
backbone
argument.- Parameters
enable_ort¶ (
bool
) – Enable Torch ONNX Runtime Optimization: https://onnxruntime.ai/docs/#onnx-runtime-for-training
- classmethod available_finetuning_strategies(cls)¶
Returns a list containing the keys of the available Finetuning Strategies.
- classmethod available_lr_schedulers(cls)¶
Returns a list containing the keys of the available LR schedulers.
- classmethod available_optimizers(cls)¶
Returns a list containing the keys of the available Optimizers.
- classmethod available_outputs(cls)¶
Returns the list of available outputs (that can be used during prediction or serving) for this
Task
.Examples
..testsetup:
>>> from flash import Task
>>> print(Task.available_outputs()) ['preds', 'raw']