This commit is contained in:
SanjaESC
2020-07-14 16:09:19 +02:00
committed by thllwg
parent 2a840222da
commit 564fc0aab4
2 changed files with 2 additions and 4 deletions
+1 -2
View File
@@ -177,8 +177,7 @@ class TacotronAbstract(ABC, nn.Module):
elif style_input is None:
gst_outputs = torch.zeros(1, 1, self.gst_embedding_dim).to(device)
else:
# pylint: disable=not-callable
gst_outputs = self.gst_layer(style_input)
gst_outputs = self.gst_layer(style_input) # pylint: disable=not-callable
embedded_gst = gst_outputs.repeat(1, inputs.size(1), 1)
return inputs, embedded_gst
+1 -2
View File
@@ -60,6 +60,5 @@
"gst_embedding_dim": 512,
"gst_num_heads": 4,
"gst_style_tokens": 10
},
}
}