rust_bert::models::marian

Type Alias MarianConfig

Source 
pub type MarianConfig = BartConfig;
Expand description

§Marian model configuration

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

Aliased Type§

struct MarianConfig {
Show 37 fields
    pub num_labels: Option<i64>,
    pub activation_function: Option<Activation>,
    pub activation_dropout: f64,
    pub attention_dropout: f64,
    pub classif_dropout: Option<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 forced_bos_token_id: Option<i64>,
    pub forced_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,

}

Fields§

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