WindyMadman/so-vits-svc
1
0
Fork 0
mirror of https://github.com/svc-develop-team/so-vits-svc.git synced 2025-01-08 11:57:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(fcpe) : windows size 8 to 9

Browse Source 
fix(fcpe) : windows size 8 to 9
This commit is contained in:
YuriHead 2023-07-26 22:22:37 +08:00 committed by GitHub
parent d8c91d675b
commit bb610c5805
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/F0Predictor/fcpe/model.py
View File

@ -136,7 +136,7 @@ class FCPE(nn.Module):
B, N, _ = y.size()
ci = self.cent_table[None, None, :].expand(B, N, -1)
confident, max_index = torch.max(y, dim=-1, keepdim=True)
local_argmax_index = torch.arange(0,8).to(max_index.device) + (max_index - 4)
local_argmax_index = torch.arange(0,9).to(max_index.device) + (max_index - 4)
local_argmax_index[local_argmax_index<0] = 0
local_argmax_index[local_argmax_index>=self.n_out] = self.n_out - 1
ci_l = torch.gather(ci,-1,local_argmax_index)