HuggingFaceBertSentenceEncoder.Config

Component: HuggingFaceBertSentenceEncoder

class HuggingFaceBertSentenceEncoder.Config

Bases: TransformerSentenceEncoderBase.Config, ConfigBase

All Attributes (including base classes)

load_path: Optional[str] = None

save_path: Optional[str] = None

freeze: bool = False

shared_module_key: Optional[str] = None

output_dropout: float = 0.4

embedding_dim: int = 768

pooling: PoolingMethod = <PoolingMethod.CLS_TOKEN: 'cls_token'>

export: bool = False

projection_dim: int = 0

normalize_output_rep: bool = False

bert_cpt_dir: str = 'manifold://nlp_technologies/tree/huggingface-models/bert-base-uncased/'

load_weights: bool = True

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "output_dropout": 0.4,
    "embedding_dim": 768,
    "pooling": "cls_token",
    "export": false,
    "projection_dim": 0,
    "normalize_output_rep": false,
    "bert_cpt_dir": "manifold://nlp_technologies/tree/huggingface-models/bert-base-uncased/",
    "load_weights": true
}