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