pub struct SqueezeAlbertConfig {
Show 20 fields pub attention_probs_dropout_prob: f64, pub embedding_size: i64, pub hidden_act: Activation, pub hidden_dropout_prob: f64, pub hidden_size: i64, pub initializer_range: f64, pub inner_group_num: i64, pub intermediate_size: i64, pub max_position_embeddings: i64, pub num_attention_heads: i64, pub num_hidden_groups: i64, pub num_hidden_layers: i64, pub type_vocab_size: i64, pub vocab_size: i64, pub q_groups: i64, pub k_groups: i64, pub v_groups: i64, pub post_attention_groups: i64, pub intermediate_groups: i64, pub output_groups: i64,
}
Expand description

SqueezeALBERT model configuration.

This is the union set of the SqueezeBERT and ALBERT configurations:

  • SqueezeBERT uses q_groups, k_groups, v_groups, post_attention_groups, intermediate_groups, and output_groups to configure the number of groups in grouped convolutions.
  • ALBERT uses num_hidden_groups to configure the number of layer groups and embedding_size to configure the size of piece embeddings.

Fields

attention_probs_dropout_prob: f64embedding_size: i64hidden_act: Activationhidden_dropout_prob: f64hidden_size: i64initializer_range: f64inner_group_num: i64intermediate_size: i64max_position_embeddings: i64num_attention_heads: i64num_hidden_groups: i64num_hidden_layers: i64type_vocab_size: i64vocab_size: i64q_groups: i64k_groups: i64v_groups: i64post_attention_groups: i64intermediate_groups: i64output_groups: i64

Trait Implementations

Formats the value using the given formatter. Read more

Returns the “default value” for a type. Read more

Deserialize this value from the given Serde deserializer. Read more

Converts to this type from the input type.

Converts to this type from the input type.

Converts to this type from the input type.

Auto Trait Implementations

Blanket Implementations

Gets the TypeId of self. Read more

Immutably borrows from an owned value. Read more

Mutably borrows from an owned value. Read more

Returns the argument unchanged.

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Should always be Self

The type returned in the event of a conversion error.

Performs the conversion.

The type returned in the event of a conversion error.

Performs the conversion.