[−][src]Struct rust_bert::bart::BartConfig

pub struct BartConfig {
    pub num_labels: Option<i64>,
    pub activation_function: Option<Activation>,
    pub activation_dropout: f64,
    pub attention_dropout: f64,
    pub classif_dropout: f64,
    pub d_model: i64,
    pub decoder_attention_heads: i64,
    pub decoder_ffn_dim: i64,
    pub decoder_layerdrop: f64,
    pub decoder_layers: i64,
    pub decoder_start_token_id: Option<i64>,
    pub dropout: f64,
    pub encoder_attention_heads: i64,
    pub encoder_ffn_dim: i64,
    pub encoder_layerdrop: f64,
    pub encoder_layers: i64,
    pub bos_token_id: Option<i64>,
    pub eos_token_id: Option<i64>,
    pub pad_token_id: Option<i64>,
    pub id2label: Option<HashMap<i64, String>>,
    pub label2id: Option<HashMap<String, i64>>,
    pub init_std: f64,
    pub is_decoder: Option<bool>,
    pub is_encoder_decoder: Option<bool>,
    pub max_position_embeddings: i64,
    pub min_length: Option<i64>,
    pub no_repeat_ngram_size: Option<i64>,
    pub normalize_embedding: Option<bool>,
    pub num_hidden_layers: i64,
    pub output_attentions: Option<bool>,
    pub output_hidden_states: Option<bool>,
    pub output_past: Option<bool>,
    pub static_position_embeddings: Option<bool>,
    pub scale_embedding: Option<bool>,
    pub vocab_size: i64,
}

BART model configuration

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

Fields

num_labels: Option<i64>activation_function: Option<Activation>activation_dropout: f64attention_dropout: f64classif_dropout: f64d_model: i64decoder_attention_heads: i64decoder_ffn_dim: i64decoder_layerdrop: f64decoder_layers: i64decoder_start_token_id: Option<i64>dropout: f64encoder_attention_heads: i64encoder_ffn_dim: i64encoder_layerdrop: f64encoder_layers: i64bos_token_id: Option<i64>eos_token_id: Option<i64>pad_token_id: Option<i64>id2label: Option<HashMap<i64, String>>label2id: Option<HashMap<String, i64>>init_std: f64is_decoder: Option<bool>is_encoder_decoder: Option<bool>max_position_embeddings: i64min_length: Option<i64>no_repeat_ngram_size: Option<i64>normalize_embedding: Option<bool>num_hidden_layers: i64output_attentions: Option<bool>output_hidden_states: Option<bool>output_past: Option<bool>static_position_embeddings: Option<bool>scale_embedding: Option<bool>vocab_size: i64

Trait Implementations

impl Config<BartConfig> for BartConfig[src]

impl Debug for BartConfig[src]

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

impl Serialize for BartConfig[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.