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