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) – 注意力分數的維度。