torchaudio.models.squim_subjective_model

torchaudio.models.squim_subjective_model(ssl_type: str, feat_dim: int, proj_dim: int, att_dim: int) → SquimSubjective

构建一个自定义的 torchaudio.prototype.models.SquimSubjective 模型。

参数:

• ssl_type (str) – 自监督学习 (SSL) 模型类型。必须是 ["wav2vec2_base", "wav2vec2_large"] 之一。

• feat_dim (int) – SSL 特征表示的特征维度。

• proj_dim (int) – 投影层的输出维度。

• att_dim (int) – 注意力分数的维度。