Save speakers embeddings/ids before starting training

This commit is contained in:
Edresson 2021-08-12 07:52:44 -03:00 committed by Eren Gölge
parent bce143c738
commit 8310d19da8
2 changed files with 1 additions and 2 deletions

View File

@ -402,7 +402,6 @@ class Vits(BaseTTS):
# speaker embedding # speaker embedding
if self.num_speakers > 1 and sid is not None: if self.num_speakers > 1 and sid is not None:
g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1] g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1]
# posterior encoder # posterior encoder
z, m_q, logs_q, y_mask = self.posterior_encoder(y, y_lengths, g=g) z, m_q, logs_q, y_mask = self.posterior_encoder(y, y_lengths, g=g)

View File

@ -13,7 +13,7 @@ pyyaml
scipy>=0.19.0 scipy>=0.19.0
soundfile soundfile
tensorboardX tensorboardX
torch>=1.7 torch>=1.9.0
tqdm tqdm
numba==0.53 numba==0.53
umap-learn==0.5.1 umap-learn==0.5.1