wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #581 from Edresson/dev

Browse Source 
Compute speaker embeddings in batch for the LSTM  Speaker Encoder and Compute embeddings/ finding chars using config file.
This commit is contained in:
Eren Gölge
2021-07-23 17:22:51 +02:00
committed by GitHub
parent 9bb7f31f36 d5adc35fdf
commit 30eed347b6
9 changed files with 106 additions and 591 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_speaker_encoder.py
+1 -1
View File
@@ -35,7 +35,7 @@ class LSTMSpeakerEncoderTests(unittest.TestCase):
assert abs(assert_diff) < 1e-4, f" [!] output_norm has wrong values - {assert_diff}"
# compute d for a given batch
dummy_input = T.rand(1, 240, 80) # B x T x D
output = model.compute_embedding(dummy_input, num_frames=160, overlap=0.5)
output = model.compute_embedding(dummy_input, num_frames=160, num_eval=5)
assert output.shape[0] == 1
assert output.shape[1] == 256
assert len(output.shape) == 2