AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

handle the exception that remove_length==1 and self.causal is false

Browse Source
This commit is contained in:
hp 2021-11-29 13:45:56 +08:00
parent ba6919baaf
commit 262ca57133
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kws/model/mdtc.py
View File

@ -249,11 +249,12 @@ class MDTC(nn.Module):
output_size = outputs_list[-1].shape[-1] output_size = outputs_list[-1].shape[-1]
for x in outputs_list: for x in outputs_list:
remove_length = x.shape[-1] - output_size remove_length = x.shape[-1] - output_size
if not self.causal:
remove_length = remove_length // 2
if remove_length != 0: if remove_length != 0:
if self.causal: if self.causal:
normalized_outputs.append(x[:, :, remove_length:]) normalized_outputs.append(x[:, :, remove_length:])
else: else:
remove_length = remove_length // 2
normalized_outputs.append(x[:, :, normalized_outputs.append(x[:, :,
remove_length:-remove_length]) remove_length:-remove_length])
else: else: