mirror of https://github.com/coqui-ai/TTS.git
27 lines
882 B
Python
27 lines
882 B
Python
import numpy as np
|
|
import torch
|
|
|
|
from TTS.vocoder.models.melgan_discriminator import MelganDiscriminator
|
|
from TTS.vocoder.models.melgan_multiscale_discriminator import MelganMultiscaleDiscriminator
|
|
|
|
|
|
def test_melgan_discriminator():
|
|
model = MelganDiscriminator()
|
|
print(model)
|
|
dummy_input = torch.rand((4, 1, 256 * 10))
|
|
output, _ = model(dummy_input)
|
|
assert np.all(output.shape == (4, 1, 10))
|
|
|
|
|
|
def test_melgan_multi_scale_discriminator():
|
|
model = MelganMultiscaleDiscriminator()
|
|
print(model)
|
|
dummy_input = torch.rand((4, 1, 256 * 16))
|
|
scores, feats = model(dummy_input)
|
|
assert len(scores) == 3
|
|
assert len(scores) == len(feats)
|
|
assert np.all(scores[0].shape == (4, 1, 64))
|
|
assert np.all(feats[0][0].shape == (4, 16, 4096))
|
|
assert np.all(feats[0][1].shape == (4, 64, 1024))
|
|
assert np.all(feats[0][2].shape == (4, 256, 256))
|