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

wavegrad refactoring, fixing tests for glow-tts and wavegrad

Browse Source
This commit is contained in:
erogol
2020-10-29 15:47:15 +01:00
parent 946a0c0fb9
commit 39c71ee8a9
6 changed files with 49 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_wavegrad_train.py
+10 -1
View File
@@ -30,9 +30,18 @@ class WavegradTrainTest(unittest.TestCase):
upsample_dilations=[[1, 2, 1, 2], [1, 2, 1, 2],
[1, 2, 4, 8], [1, 2, 4, 8],
[1, 2, 4, 8]])
model_ref = Wavegrad(in_channels=80,
out_channels=1,
upsample_factors=[5, 5, 3, 2, 2],
upsample_dilations=[[1, 2, 1, 2], [1, 2, 1, 2],
[1, 2, 4, 8], [1, 2, 4, 8],
[1, 2, 4, 8]])
model.train()
model.to(device)
model_ref = copy.deepcopy(model)
model.compute_noise_level(1000, 1e-6, 1e-2)
model_ref.load_state_dict(model.state_dict())
model_ref.to(device)
count = 0
for param, param_ref in zip(model.parameters(),
model_ref.parameters()):