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 的块输入步长。