torchaudio.models.conv_tasnet_base¶
- torchaudio.models.conv_tasnet_base(num_sources: int = 2) ConvTasNet [源代码]¶
构建
ConvTasNet
的非因果版本。参数设置遵循论文中 Si-SNR 指标得分最高的设置,除了掩码激活函数从“sigmoid”更改为“relu”以提高性能。
- 参数:
num_sources (int, 可选) – 输出中的源数量。(默认值:2)
- 返回值:
ConvTasNet 模型。
- 返回类型: