[−][src]Struct rust_bert::distilbert::DistilBertConfig

pub struct DistilBertConfig {
    pub activation: Activation,
    pub attention_dropout: f64,
    pub dim: i64,
    pub dropout: f64,
    pub hidden_dim: i64,
    pub id2label: Option<HashMap<i64, String>>,
    pub initializer_range: f32,
    pub is_decoder: Option<bool>,
    pub label2id: Option<HashMap<String, i64>>,
    pub max_position_embeddings: i64,
    pub n_heads: i64,
    pub n_layers: i64,
    pub output_attentions: Option<bool>,
    pub output_hidden_states: Option<bool>,
    pub output_past: Option<bool>,
    pub qa_dropout: f64,
    pub seq_classif_dropout: f64,
    pub sinusoidal_pos_embds: bool,
    pub tie_weights_: bool,
    pub torchscript: Option<bool>,
    pub use_bfloat16: Option<bool>,
    pub vocab_size: i64,
}

DistilBERT model configuration

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

Fields

activation: Activationattention_dropout: f64dim: i64dropout: f64hidden_dim: i64id2label: Option<HashMap<i64, String>>initializer_range: f32is_decoder: Option<bool>label2id: Option<HashMap<String, i64>>max_position_embeddings: i64n_heads: i64n_layers: i64output_attentions: Option<bool>output_hidden_states: Option<bool>output_past: Option<bool>qa_dropout: f64seq_classif_dropout: f64sinusoidal_pos_embds: booltie_weights_: booltorchscript: Option<bool>use_bfloat16: Option<bool>vocab_size: i64

Trait Implementations

impl Config<DistilBertConfig> for DistilBertConfig[src]

impl Debug for DistilBertConfig[src]

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

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