[][src]Struct rust_bert::t5::T5Config

pub struct T5Config {
    pub dropout_rate: f64,
    pub d_model: i64,
    pub d_ff: i64,
    pub d_kv: i64,
    pub decoder_start_token_id: Option<i64>,
    pub eos_token_id: Option<i64>,
    pub initializer_factor: f64,
    pub is_encoder_decoder: Option<bool>,
    pub layer_norm_epsilon: f64,
    pub n_positions: i64,
    pub num_heads: i64,
    pub num_layers: i64,
    pub output_past: Option<bool>,
    pub pad_token_id: Option<i64>,
    pub relative_attention_num_buckets: i64,
    pub vocab_size: i64,
    // some fields omitted
}

T5 model configuration

Defines the T5 model architecture (e.g. number of layers, hidden layer size, label mapping...)

Fields

dropout_rate: f64d_model: i64d_ff: i64d_kv: i64decoder_start_token_id: Option<i64>eos_token_id: Option<i64>initializer_factor: f64is_encoder_decoder: Option<bool>layer_norm_epsilon: f64n_positions: i64num_heads: i64num_layers: i64output_past: Option<bool>pad_token_id: Option<i64>relative_attention_num_buckets: i64vocab_size: i64

Trait Implementations

impl Config<T5Config> for T5Config[src]

impl Debug for T5Config[src]

impl<'de> Deserialize<'de> for T5Config[src]

impl Serialize for T5Config[src]

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized
[src]

impl<T> Borrow<T> for T where
    T: ?Sized
[src]

impl<T> BorrowMut<T> for T where
    T: ?Sized
[src]

impl<T> DeserializeOwned for T where
    T: for<'de> Deserialize<'de>, 
[src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, 
[src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<V, T> VZip<V> for T where
    V: MultiLane<T>,