Struct rust_bert::xlnet::XLNetConfig[][src]

pub struct XLNetConfig {
    pub vocab_size: i64,
    pub d_model: i64,
    pub n_layer: i64,
    pub d_head: i64,
    pub n_head: i64,
    pub d_inner: i64,
    pub ff_activation: Activation,
    pub untie_r: bool,
    pub attn_type: AttentionType,
    pub initializer_range: f32,
    pub layer_norm_eps: Option<f64>,
    pub dropout: f64,
    pub mem_len: Option<i64>,
    pub reuse_len: Option<i64>,
    pub clamp_len: Option<i64>,
    pub bi_data: bool,
    pub same_length: bool,
    pub summary_type: Option<SummaryType>,
    pub summary_use_proj: Option<bool>,
    pub summary_activation: Option<Activation>,
    pub summary_proj_to_labels: Option<bool>,
    pub summary_first_dropout: Option<f64>,
    pub summary_last_dropout: Option<f64>,
    pub start_n_top: Option<i64>,
    pub end_n_top: Option<i64>,
    pub use_cache: Option<bool>,
    pub bos_token_id: i64,
    pub eos_token_id: i64,
    pub pad_token_id: i64,
    pub id2label: Option<HashMap<i64, String>>,
    pub label2id: Option<HashMap<String, i64>>,
    pub output_attentions: Option<bool>,
    pub output_hidden_states: Option<bool>,
    pub chunk_size_feed_forward: Option<i64>,
}

XLNet model configuration

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

Fields

vocab_size: i64d_model: i64n_layer: i64d_head: i64n_head: i64d_inner: i64ff_activation: Activationuntie_r: boolattn_type: AttentionTypeinitializer_range: f32layer_norm_eps: Option<f64>dropout: f64mem_len: Option<i64>reuse_len: Option<i64>clamp_len: Option<i64>bi_data: boolsame_length: boolsummary_type: Option<SummaryType>summary_use_proj: Option<bool>summary_activation: Option<Activation>summary_proj_to_labels: Option<bool>summary_first_dropout: Option<f64>summary_last_dropout: Option<f64>start_n_top: Option<i64>end_n_top: Option<i64>use_cache: Option<bool>bos_token_id: i64eos_token_id: i64pad_token_id: i64id2label: Option<HashMap<i64, String>>label2id: Option<HashMap<String, i64>>output_attentions: Option<bool>output_hidden_states: Option<bool>chunk_size_feed_forward: Option<i64>

Trait Implementations

impl Clone for XLNetConfig[src]

impl Config<XLNetConfig> for XLNetConfig[src]

impl Debug for XLNetConfig[src]

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

impl Serialize for XLNetConfig[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> Instrument for T[src]

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

impl<T> Pointable for T

type Init = T

The type for initializers.

impl<T> Same<T> for T

type Output = T

Should always be Self

impl<T> ToOwned for T where
    T: Clone
[src]

type Owned = T

The resulting type after obtaining ownership.

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>,