pub struct TransformerConfig {
pub input_dim: usize,
pub embed_dim: usize,
pub num_heads: usize,
pub num_encoder_layers: usize,
pub num_decoder_layers: usize,
pub feedforward_dim: usize,
pub dropout: f32,
pub max_seq_len: usize,
}Expand description
Configuration for the Transformer tokenizer
Fields§
§input_dim: usizeInput signal dimension
embed_dim: usizeEmbedding dimension (must be divisible by num_heads)
num_heads: usizeNumber of attention heads
num_encoder_layers: usizeNumber of encoder layers
num_decoder_layers: usizeNumber of decoder layers
feedforward_dim: usizeFeedforward network hidden dimension
dropout: f32Dropout probability (0.0 to 1.0)
max_seq_len: usizeMaximum sequence length for positional encoding
Implementations§
Source§impl TransformerConfig
impl TransformerConfig
Sourcepub fn validate(&self) -> TokenizerResult<()>
pub fn validate(&self) -> TokenizerResult<()>
Validate the configuration
Trait Implementations§
Source§impl Clone for TransformerConfig
impl Clone for TransformerConfig
Source§fn clone(&self) -> TransformerConfig
fn clone(&self) -> TransformerConfig
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for TransformerConfig
impl Debug for TransformerConfig
Source§impl Default for TransformerConfig
impl Default for TransformerConfig
Source§impl<'de> Deserialize<'de> for TransformerConfig
impl<'de> Deserialize<'de> for TransformerConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for TransformerConfig
impl RefUnwindSafe for TransformerConfig
impl Send for TransformerConfig
impl Sync for TransformerConfig
impl Unpin for TransformerConfig
impl UnwindSafe for TransformerConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more