Shortcuts

torchaudio.models.wav2vec2_xlsr_1b

torchaudio.models.wav2vec2_xlsr_1b(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[source]

Builds XLS-R model [Babu et al., 2021] with 1 billion of parameters. The architecture is compatible with Wav2Vec2 model [Baevski et al., 2020], and so the output class is Wav2Vec2Model.

Parameters:
Returns:

The resulting model.

Return type:

Wav2Vec2Model

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources