torchaudio.models.wav2vec2_xlsr_2b

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

构建具有 20 亿参数的 XLS-R 模型[Babu 等人，2021]。该架构与 Wav2Vec2 模型[Baevski 等人，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, 可选) – 请参阅wav2vec2_model()。

返回值:

生成的模型。

返回类型:

Wav2Vec2Model