| funasr/models/frontend/default.py | ●●●●● 补丁 | 查看 | 原始文档 | 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))