File size: 939 Bytes
9bbba62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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