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, 選填) – 編碼器模組後的特徵維度。

• 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 或 None, 選填) – DPRNN 區塊輸入的步幅。