yhliang
2023-05-18 1e650fac78bf03ed40051cc41fad9fa16c236e3d
fix bug
1个文件已修改
4 ■■■■ 已修改文件
funasr/models/frontend/default.py 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/frontend/default.py
@@ -102,8 +102,8 @@
        if input_stft.dim() == 4:
            # h: (B, T, C, F) -> h: (B, T, F)
            if self.training:
                if self.use_channel == None:
                    input_stft = input_stft[:, :, 0, :]
                if self.use_channel is not None:
                    input_stft = input_stft[:, :, self.use_channel, :]
                else:
                    # Select 1ch randomly
                    ch = np.random.randint(input_stft.size(2))