Struct EncoderConfig 

pub struct EncoderConfig {
    pub attention: AttentionConfig,
    pub feed_forward: FeedForwardConfig,
    pub layer_norm: LayerNormConfig,
    pub pre_norm: bool,
}

Configuration for transformer encoder layer

Fields

attention: AttentionConfig

Attention configuration

feed_forward: FeedForwardConfig

Feed-forward configuration

layer_norm: LayerNormConfig

Layer normalization configuration

pre_norm: bool

Whether to use pre-layer normalization (vs post)

Implementations

impl EncoderConfig

pub fn new(d_model: usize, n_heads: usize, d_ff: usize) -> Result<Self>

Create a new encoder configuration

pub fn with_pre_norm(self, pre_norm: bool) -> Self

Set pre-normalization vs post-normalization

pub fn with_causal(self, causal: bool) -> Self

Set causal masking

pub fn with_dropout(self, dropout: f64) -> Self

Set dropout

pub fn validate(&self) -> Result<()>

Validate configuration

Trait Implementations

impl Clone for EncoderConfig

fn clone(&self) -> EncoderConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EncoderConfig

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

Formats the value using the given formatter.