torchaudio.models.squim_objective_model

torchaudio.models.squim_objective_model(feat_dim: int, win_len: int, d_model: int, nhead: int, hidden_dim: int, num_blocks: int, rnn_type: str, chunk_size: int, chunk_stride: Optional[int] = None) → SquimObjective

構建自定義的 torchaudio.prototype.models.SquimObjective 模型。

引數：

• feat_dim (int, optional) – 編碼器模組後的特徵維度。

• win_len (int) – 編碼器模組中的核大小。

• d_model (int) – 輸入中預期的特徵數量。

• nhead (int) – 多頭注意力模型中的頭數量。

• hidden_dim (int) – DPRNN 的 RNN 層中的隱藏維度。

• num_blocks (int) – DPRNN 層的數量。

• rnn_type (str) – DPRNN 中 RNN 的型別。有效選項為 ["RNN", "LSTM", "GRU"]。

• chunk_size (int) – DPRNN 輸入的塊大小。

• chunk_stride (int or None, optional) – DPRNN 塊輸入的步幅。