Struct Wav2Vec2BertConfig 

pub struct Wav2Vec2BertConfig {

    pub hidden_size: usize,
    pub num_hidden_layers: usize,
    pub num_attention_heads: usize,
    pub intermediate_size: usize,
    pub feature_projection_input_dim: usize,
    pub layer_norm_eps: f64,
    pub hidden_act: String,
    pub position_embeddings_type: String,
    pub left_max_position_embeddings: usize,
    pub right_max_position_embeddings: usize,
    pub conv_depthwise_kernel_size: usize,
    pub add_adapter: bool,
    pub apply_spec_augment: bool,
    pub use_intermediate_ffn_before_adapter: bool,
    pub model_type: Option<String>,
}

Wav2Vec2-BERT model configuration (e.g. facebook/w2v-bert-2.0).

Fields

hidden_size: usize

num_hidden_layers: usize

num_attention_heads: usize

intermediate_size: usize

feature_projection_input_dim: usize

layer_norm_eps: f64

hidden_act: String

position_embeddings_type: String

left_max_position_embeddings: usize

right_max_position_embeddings: usize

conv_depthwise_kernel_size: usize

add_adapter: bool

apply_spec_augment: bool

use_intermediate_ffn_before_adapter: bool

model_type: Option<String>

Present in HF configs; ignored at inference when apply_spec_augment=false.

Implementations

impl Wav2Vec2BertConfig

pub fn from_file(path: &Path) -> Result<Wav2Vec2BertConfig, Error>

pub fn head_dim(&self) -> usize

pub fn num_relative_positions(&self) -> usize

pub fn w2v_bert_2_0() -> Wav2Vec2BertConfig

Factory for the public W2v-BERT 2.0 checkpoint dimensions.

Trait Implementations

impl Clone for Wav2Vec2BertConfig

fn clone(&self) -> Wav2Vec2BertConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Wav2Vec2BertConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Wav2Vec2BertConfig

fn deserialize<__D>( __deserializer: __D, ) -> Result<Wav2Vec2BertConfig, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.