torchaudio.models.wav2vec2_base

torchaudio.models.wav2vec2_base(encoder_projection_dropout: float = 0.1, encoder_attention_dropout: float = 0.1, encoder_ff_interm_dropout: float = 0.1, encoder_dropout: float = 0.1, encoder_layer_drop: float = 0.1, aux_num_out: Optional[int] = None) → Wav2Vec2Model

構建 wav2vec 2.0 [Baevski et al., 2020] 中的“base” Wav2Vec2Model 模型。

引數：

• encoder_projection_dropout (float) – 參見 wav2vec2_model()。

• encoder_attention_dropout (float) – 參見 wav2vec2_model()。

• encoder_ff_interm_dropout (float) – 參見 wav2vec2_model()。

• encoder_dropout (float) – 參見 wav2vec2_model()。

• encoder_layer_drop (float) – 參見 wav2vec2_model()。

• aux_num_out (int 或 None, 可選) – 參見 wav2vec2_model()。

返回值：

返回的模型例項。

返回型別：

Wav2Vec2Model