This commit is contained in:
lyuxiang.lx
2025-04-07 23:04:01 +08:00
parent cfc68f379c
commit f0b8e892f6

View File

@@ -165,7 +165,7 @@ class MultiResSpecDiscriminator(torch.nn.Module):
y_d_gs = [] y_d_gs = []
fmap_rs = [] fmap_rs = []
fmap_gs = [] fmap_gs = []
for i, d in enumerate(self.discriminators): for _, d in enumerate(self.discriminators):
y_d_r, fmap_r = d(y) y_d_r, fmap_r = d(y)
y_d_g, fmap_g = d(y_hat) y_d_g, fmap_g = d(y_hat)
y_d_rs.append(y_d_r) y_d_rs.append(y_d_r)
@@ -219,7 +219,7 @@ class SpecDiscriminator(nn.Module):
y = y.squeeze(1) y = y.squeeze(1)
y = stft(y, self.fft_size, self.shift_size, self.win_length, self.window.to(y.device)) y = stft(y, self.fft_size, self.shift_size, self.win_length, self.window.to(y.device))
y = y.unsqueeze(1) y = y.unsqueeze(1)
for i, d in enumerate(self.discriminators): for _, d in enumerate(self.discriminators):
y = d(y) y = d(y)
y = F.leaky_relu(y, LRELU_SLOPE) y = F.leaky_relu(y, LRELU_SLOPE)
fmap.append(y) fmap.append(y)