Struct BertConfig 

pub struct BertConfig {
    pub vocab_size: usize,
    pub hidden_size: usize,
    pub num_layers: usize,
    pub num_heads: usize,
    pub intermediate_size: usize,
    pub max_position_embeddings: usize,
    pub type_vocab_size: usize,
    pub dropout: f32,
    pub layer_norm_eps: f32,
}

BERT configuration

Fields

vocab_size: usize

Vocabulary size

hidden_size: usize

Hidden size (embedding dimension)

num_layers: usize

Number of transformer layers

num_heads: usize

Number of attention heads

intermediate_size: usize

Intermediate size in feed-forward network

max_position_embeddings: usize

Maximum sequence length

type_vocab_size: usize

Number of token types (segments)

dropout: f32

Dropout probability

layer_norm_eps: f32

Layer norm epsilon

Implementations

impl BertConfig

pub fn bert_base() -> Self

BERT-Base configuration

pub fn bert_large() -> Self

BERT-Large configuration

pub fn bert_tiny() -> Self

BERT-Tiny configuration (for testing)

Trait Implementations

impl Clone for BertConfig

fn clone(&self) -> BertConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BertConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for BertConfig

fn default() -> Self

Returns the “default value” for a type.