mirror of
https://github.com/wassname/TTS.git
synced 2026-10-02 12:00:11 +08:00
reformatting and styling
This commit is contained in:
1 parent
9011dddf77
commit
f519012dea
159 files changed
+6589
-6429
No files matched your search
@@ -1,7 +1,10 @@
|
||||
import numpy as np
|
||||
import torch
|
||||
|
||||
from TTS.vocoder.models.parallel_wavegan_discriminator import ParallelWaveganDiscriminator, ResidualParallelWaveganDiscriminator
|
||||
from TTS.vocoder.models.parallel_wavegan_discriminator import (
|
||||
ParallelWaveganDiscriminator,
|
||||
ResidualParallelWaveganDiscriminator,
|
||||
)
|
||||
|
||||
|
||||
def test_pwgan_disciminator():
|
||||
@@ -14,7 +17,8 @@ def test_pwgan_disciminator():
|
||||
dilation_factor=1,
|
||||
nonlinear_activation="LeakyReLU",
|
||||
nonlinear_activation_params={"negative_slope": 0.2},
|
||||
bias=True)
|
||||
bias=True,
|
||||
)
|
||||
dummy_x = torch.rand((4, 1, 64 * 256))
|
||||
output = model(dummy_x)
|
||||
assert np.all(output.shape == (4, 1, 64 * 256))
|
||||
@@ -34,7 +38,8 @@ def test_redisual_pwgan_disciminator():
|
||||
dropout=0.0,
|
||||
bias=True,
|
||||
nonlinear_activation="LeakyReLU",
|
||||
nonlinear_activation_params={"negative_slope": 0.2})
|
||||
nonlinear_activation_params={"negative_slope": 0.2},
|
||||
)
|
||||
dummy_x = torch.rand((4, 1, 64 * 256))
|
||||
output = model(dummy_x)
|
||||
assert np.all(output.shape == (4, 1, 64 * 256))
|
||||
|
||||
Reference in new issue
Block a user