Back to Sentence Transformers

Modules

docs/package_reference/base/modules.rst

5.4.1938 B
Original Source

Modules

sentence_transformers.base.modules defines different building blocks, a.k.a. Modules, that can be used to create models from scratch.

Common Modules

.. autoclass:: sentence_transformers.base.modules.Transformer :members: modalities, preprocess, modality_config, transformer_task, max_seq_length, get_embedding_dimension

.. autoclass:: sentence_transformers.base.modules.Dense

.. autoclass:: sentence_transformers.base.modules.Router :members: for_query_document

Base Modules

.. autoclass:: sentence_transformers.base.modules.Module :members: config_file_name, config_keys, save_in_root, forward, get_config_dict, load, load_config, load_file_path, load_dir_path, load_torch_weights, save, save_config, save_torch_weights

.. autoclass:: sentence_transformers.base.modules.InputModule :members: save_in_root, tokenizer, modalities, preprocess, tokenize, save_tokenizer