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

fix flake8

Browse Source
This commit is contained in:
menglong.xu 2023-09-01 19:45:26 +08:00
parent 5074b61182
commit 19e0c228b1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wekws/model/fsmn.py
View File

@ -478,7 +478,7 @@ class FSMN(nn.Module):
in_cache = [torch.zeros(0, 0, 0, 0, dtype=torch.float) in_cache = [torch.zeros(0, 0, 0, 0, dtype=torch.float)
for _ in range(len(self.fsmn))] for _ in range(len(self.fsmn))]
else: else:
in_cache = [in_cache[:, :, :, i:i+1] for i in range(in_cache.size(-1))] in_cache = [in_cache[:, :, :, i: i+1] for i in range(in_cache.size(-1))]
input = (input, in_cache) input = (input, in_cache)
x1 = self.in_linear1(input) x1 = self.in_linear1(input)
x2 = self.in_linear2(x1) x2 = self.in_linear2(x1)