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