pub struct SqueezeBertConfig {Show 19 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 intermediate_size: i64,
pub layer_norm_eps: f64,
pub max_position_embeddings: i64,
pub num_attention_heads: 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
SqueezeBert model configuration.
Fields
attention_probs_dropout_prob: f64
embedding_size: i64
initializer_range: f64
intermediate_size: i64
layer_norm_eps: f64
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 SqueezeBertConfig
impl Debug for SqueezeBertConfig
sourceimpl Default for SqueezeBertConfig
impl Default for SqueezeBertConfig
sourceimpl<'de> Deserialize<'de> for SqueezeBertConfig where
SqueezeBertConfig: Default,
impl<'de> Deserialize<'de> for SqueezeBertConfig where
SqueezeBertConfig: 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 SqueezeBertConfig
impl From<&SqueezeAlbertConfig> for SqueezeBertConfig
sourcefn from(config: &SqueezeAlbertConfig) -> Self
fn from(config: &SqueezeAlbertConfig) -> Self
Converts to this type from the input type.
sourceimpl From<&SqueezeBertConfig> for BertConfig
impl From<&SqueezeBertConfig> for BertConfig
sourcefn from(squeeze_bert_config: &SqueezeBertConfig) -> Self
fn from(squeeze_bert_config: &SqueezeBertConfig) -> Self
Converts to this type from the input type.
Auto Trait Implementations
impl RefUnwindSafe for SqueezeBertConfig
impl Send for SqueezeBertConfig
impl Sync for SqueezeBertConfig
impl Unpin for SqueezeBertConfig
impl UnwindSafe for SqueezeBertConfig
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