[−][src]Struct rust_bert::gpt2::Gpt2Config

pub struct Gpt2Config {
    pub attn_pdrop: Option<f64>,
    pub embd_pdrop: Option<f64>,
    pub hidden_dropout_prob: Option<f64>,
    pub afn: Option<GptActivation>,
    pub initializer_range: f64,
    pub layer_norm_epsilon: f64,
    pub n_ctx: i64,
    pub n_embd: i64,
    pub n_head: i64,
    pub n_layer: i64,
    pub n_positions: i64,
    pub num_labels: Option<i64>,
    pub output_past: Option<bool>,
    pub output_attentions: Option<bool>,
    pub output_hidden_states: Option<bool>,
    pub resid_pdrop: Option<f64>,
    pub vocab_size: i64,
}

GPT2 model configuration

Defines the GPT2 model architecture (e.g. number of layers, hidden layer size, vocab size...). Shared between GPT and GPT2 models

Fields

attn_pdrop: Option<f64>embd_pdrop: Option<f64>hidden_dropout_prob: Option<f64>afn: Option<GptActivation>initializer_range: f64layer_norm_epsilon: f64n_ctx: i64n_embd: i64n_head: i64n_layer: i64n_positions: i64num_labels: Option<i64>output_past: Option<bool>output_attentions: Option<bool>output_hidden_states: Option<bool>resid_pdrop: Option<f64>vocab_size: i64

Trait Implementations

impl Config<Gpt2Config> for Gpt2Config[src]

impl Debug for Gpt2Config[src]

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

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