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
, andoutput_groups
to configure the number of groups in grouped convolutions. - ALBERT uses
num_hidden_groups
to configure the number of layer groups andembedding_size
to configure the size of piece embeddings.
Fields
attention_probs_dropout_prob: f64
embedding_size: i64
initializer_range: f64
inner_group_num: i64
intermediate_size: i64
max_position_embeddings: i64
num_attention_heads: i64
type_vocab_size: i64
vocab_size: i64
q_groups: i64
k_groups: i64
v_groups: i64
post_attention_groups: i64
intermediate_groups: i64
output_groups: i64
Trait Implementations
sourceimpl Debug for SqueezeAlbertConfig
impl Debug for SqueezeAlbertConfig
sourceimpl Default for SqueezeAlbertConfig
impl Default for SqueezeAlbertConfig
sourceimpl<'de> Deserialize<'de> for SqueezeAlbertConfig where
SqueezeAlbertConfig: Default,
impl<'de> Deserialize<'de> for SqueezeAlbertConfig where
SqueezeAlbertConfig: Default,
sourcefn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
sourceimpl From<&SqueezeAlbertConfig> for AlbertConfig
impl From<&SqueezeAlbertConfig> for AlbertConfig
sourcefn from(albert_config: &SqueezeAlbertConfig) -> Self
fn from(albert_config: &SqueezeAlbertConfig) -> Self
Converts to this type from the input type.
sourceimpl From<&SqueezeAlbertConfig> for BertConfig
impl From<&SqueezeAlbertConfig> for BertConfig
sourcefn from(albert_config: &SqueezeAlbertConfig) -> Self
fn from(albert_config: &SqueezeAlbertConfig) -> Self
Converts to this type from the input type.
sourceimpl From<&SqueezeAlbertConfig> for SqueezeBertConfig
impl From<&SqueezeAlbertConfig> for SqueezeBertConfig
sourcefn from(config: &SqueezeAlbertConfig) -> Self
fn from(config: &SqueezeAlbertConfig) -> Self
Converts to this type from the input type.
Auto Trait Implementations
impl RefUnwindSafe for SqueezeAlbertConfig
impl Send for SqueezeAlbertConfig
impl Sync for SqueezeAlbertConfig
impl Unpin for SqueezeAlbertConfig
impl UnwindSafe for SqueezeAlbertConfig
Blanket Implementations
sourceimpl<T> BorrowMut<T> for T where
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
const: unstable · sourcefn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more