eden-emu/opus
15
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bugfix

Browse source
This commit is contained in:
Jan Buethe 2023-09-22 11:39:22 +02:00
parent aad74fdfd6
commit 00580a63aa
No known key found for this signature in database
GPG key ID: 9E32027A35B36314
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
dnn/torch/osce/models/lavoce.py
View file

@ -89,6 +89,7 @@ class LaVoce(nn.Module):
self.kernel_size = kernel_size self.kernel_size = kernel_size
self.preemph = preemph self.preemph = preemph
self.pulses = pulses self.pulses = pulses
self.ftrans_k = ftrans_k
assert self.FEATURE_FRAME_SIZE % self.FRAME_SIZE == 0 assert self.FEATURE_FRAME_SIZE % self.FRAME_SIZE == 0
self.upsamp_factor = self.FEATURE_FRAME_SIZE // self.FRAME_SIZE self.upsamp_factor = self.FEATURE_FRAME_SIZE // self.FRAME_SIZE
@ -145,7 +146,7 @@ class LaVoce(nn.Module):
f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1) f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1)
if self.pulses: if self.pulses:
alpha = torch.cos(f) alpha = torch.cos(f).view(batch_size, 1, 1)
chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE) chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE)
pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha) pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha)
pulse_b = torch.relu(-chunk_sin - alpha) / (1 - alpha) pulse_b = torch.relu(-chunk_sin - alpha) / (1 - alpha)
@ -186,7 +187,7 @@ class LaVoce(nn.Module):
def feature_transform(self, f, layer): def feature_transform(self, f, layer):
f = f.permute(0, 2, 1) f = f.permute(0, 2, 1)
f = F.pad(f, [1, 0]) f = F.pad(f, [self.ftrans_k - 1, 0])
f = torch.tanh(layer(f)) f = torch.tanh(layer(f))
return f.permute(0, 2, 1) return f.permute(0, 2, 1)

2
dnn/torch/osce/models/lavoce_400.py
View file

@ -130,7 +130,7 @@ class LaVoce400(nn.Module):
f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1) f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1)
if self.pulses: if self.pulses:
alpha = torch.cos(f) alpha = torch.cos(f).view(batch_size, 1, 1)
chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE) chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE)
pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha) pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha)
pulse_b = torch.relu(-chunk_sin - alpha) / (1 - alpha) pulse_b = torch.relu(-chunk_sin - alpha) / (1 - alpha)