ref: aad74fdfd6ffe7df1986cfb6f410f203c16ab094
parent: 0a92bc5eaa6467d63efbed0b5ff625db64be5629
author: Jan Buethe <jbuethe@amazon.de>
date: Fri Sep 22 07:18:41 EDT 2023
pulse-related bugfix
--- a/dnn/torch/osce/models/lavoce.py
+++ b/dnn/torch/osce/models/lavoce.py
@@ -133,7 +133,7 @@
self.post_af3 = nn.Conv1d(cond_dim, cond_dim, ftrans_k)
- def create_phase_signals(self, periods, pulses=False):
+ def create_phase_signals(self, periods):
batch_size = periods.size(0)
progression = torch.arange(1, self.FRAME_SIZE + 1, dtype=periods.dtype, device=periods.device).view((1, -1))
@@ -144,7 +144,7 @@
for sframe in range(periods.size(1)):
f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1)
- if pulses:
+ if self.pulses:
alpha = torch.cos(f)
chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE)
pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha)
--- a/dnn/torch/osce/models/lavoce_400.py
+++ b/dnn/torch/osce/models/lavoce_400.py
@@ -118,7 +118,7 @@
self.post_af3 = nn.Conv1d(cond_dim, cond_dim, 2)
- def create_phase_signals(self, periods, pulses=False):
+ def create_phase_signals(self, periods):
batch_size = periods.size(0)
progression = torch.arange(1, self.FRAME_SIZE + 1, dtype=periods.dtype, device=periods.device).view((1, -1))
@@ -129,7 +129,7 @@
for sframe in range(periods.size(1)):
f = (2.0 * torch.pi / periods[:, sframe]).unsqueeze(-1)
- if pulses:
+ if self.pulses:
alpha = torch.cos(f)
chunk_sin = torch.sin(f * progression + phase0).view(batch_size, 1, self.FRAME_SIZE)
pulse_a = torch.relu(chunk_sin - alpha) / (1 - alpha)
--
⑨