mirror of https://github.com/coqui-ai/TTS.git
22 lines
816 B
Python
22 lines
816 B
Python
import torch
|
|
import numpy as np
|
|
|
|
from TTS.vocoder.models.random_window_discriminator import RandomWindowDiscriminator
|
|
|
|
|
|
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))
|