2020-05-30 16:09:25 +00:00
|
|
|
import torch
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
2020-08-04 08:51:19 +00:00
|
|
|
from mozilla_voice_tts.vocoder.models.random_window_discriminator import RandomWindowDiscriminator
|
2020-05-30 16:09:25 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_rwd():
|
|
|
|
|
layer = RandomWindowDiscriminator(cond_channels=80,
|
|
|
|
|
window_sizes=(512, 1024, 2048, 4096,
|
|
|
|
|
8192),
|
|
|
|
|
cond_disc_downsample_factors=[
|
|
|
|
|
(8, 4, 2, 2, 2), (8, 4, 2, 2),
|
|
|
|
|
(8, 4, 2), (8, 4), (4, 2, 2)
|
|
|
|
|
],
|
|
|
|
|
hop_length=256)
|
|
|
|
|
x = torch.rand([4, 1, 22050])
|
|
|
|
|
c = torch.rand([4, 80, 22050 // 256])
|
|
|
|
|
|
|
|
|
|
scores, _ = layer(x, c)
|
|
|
|
|
assert len(scores) == 10
|
|
|
|
|
assert np.all(scores[0].shape == (4, 1, 1))
|
