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