torchaudio.models.wavlm_base

torchaudio.models.wavlm_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

建構 “base” WaveLM 模型 [Chen et al., 2022]。此架構與 Wav2Vec2 模型相容 [Baevski et al., 2020]，因此輸出類別為 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, optional) – 請參閱 wav2vec2_model()。

回傳:

產生的模型。

回傳型別:

Wav2Vec2Model