[−][src]Struct rust_bert::bart::BartConfig
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: f64
attention_dropout: f64
classif_dropout: 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>
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>
output_attentions: 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]
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
__D: Deserializer<'de>,
[src]
__D: Deserializer<'de>,
impl Serialize for BartConfig
[src]
Auto Trait Implementations
impl RefUnwindSafe for BartConfig
impl Send for BartConfig
impl Sync for BartConfig
impl Unpin for BartConfig
impl UnwindSafe for BartConfig
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> DeserializeOwned for T where
T: for<'de> Deserialize<'de>,
[src]
T: for<'de> Deserialize<'de>,
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,