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