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 模型。
- 返回类型: