blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-26 15:30:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复“修复ge.sum数值可能爆炸的”可能导致的训练爆炸的问题

Browse Source 
修复“修复ge.sum数值可能爆炸的”可能导致的训练爆炸的问题
This commit is contained in:
RVC-Boss 2025-06-11 23:14:52 +08:00 committed by GitHub
parent cd6de7398e
commit ed89a02337
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
GPT_SoVITS/module/modules.py
View File

@ -1,4 +1,6 @@
import math import math
import pdb
import numpy as np import numpy as np
import torch import torch
from torch import nn from torch import nn
@ -716,11 +718,11 @@ class MelStyleEncoder(nn.Module):
if mask is None: if mask is None:
out = torch.mean(x, dim=1) out = torch.mean(x, dim=1)
else: else:
len_ = (~mask).sum() len_ = (~mask).sum(dim=1).unsqueeze(1)
x = x.masked_fill(mask.unsqueeze(-1), 0) x = x.masked_fill(mask.unsqueeze(-1), 0)
dtype=x.dtype dtype=x.dtype
x = x.float() x = x.float()
x=torch.div(x,len_) x=torch.div(x,len_.unsqueeze(1))
out=x.sum(dim=1).to(dtype) out=x.sum(dim=1).to(dtype)
return out return out