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