torchaudio.models.hubert_pretrain_base

torchaudio.models.hubert_pretrain_base(encoder_projection_dropout: float = 0.1, encoder_attention_dropout: float = 0.1, encoder_ff_interm_dropout: float = 0.0, encoder_dropout: float = 0.1, encoder_layer_drop: float = 0.05, mask_prob: float = 0.8, mask_channel_prob: float = 0.0, mask_channel_length: int = 10, feature_grad_mult: Optional[float] = 0.1, num_classes: int = 100) → HuBERTPretrainModel

构建来自 HuBERT [Hsu 等人，2021] 的“基础”HuBERTPretrainModel，用于预训练。

参数：

• encoder_projection_dropout (float) – 请参阅 hubert_pretrain_model()。

• encoder_attention_dropout (float) – 请参阅 hubert_pretrain_model()。

• encoder_ff_interm_dropout (float) – 请参阅 hubert_pretrain_model()。

• encoder_dropout (float) – 请参阅 hubert_pretrain_model()。

• encoder_layer_drop (float) – 请参阅 hubert_pretrain_model()。

• mask_prob (float) – 请参阅 hubert_pretrain_model()。

• mask_channel_prob (浮点数) – 请参见 hubert_pretrain_model()。

• mask_channel_length (整数) – 请参见 hubert_pretrain_model()。

• feature_grad_mult (浮点数 或 None) – 请参见 hubert_pretrain_model()。

• num_classes (整数, 可选) – 请参见 hubert_pretrain_model()。

返回:

生成的模型。

返回类型:

HuBERTPretrainModel