修复synthesizer/models/tacotron.Encoder注释错误 (#203)

fix Issue#202
This commit is contained in:
DragonDreamer 2021-11-09 13:59:19 +08:00 committed by GitHub
parent 6c8f3f4515
commit 4edebdfeba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -60,7 +60,7 @@ class Encoder(nn.Module):
idx = 1 idx = 1
# Start by making a copy of each speaker embedding to match the input text length # Start by making a copy of each speaker embedding to match the input text length
# The output of this has size (batch_size, num_chars * tts_embed_dims) # The output of this has size (batch_size, num_chars * speaker_embedding_size)
speaker_embedding_size = speaker_embedding.size()[idx] speaker_embedding_size = speaker_embedding.size()[idx]
e = speaker_embedding.repeat_interleave(num_chars, dim=idx) e = speaker_embedding.repeat_interleave(num_chars, dim=idx)