mirror of
https://github.com/wassname/TTS.git
synced 2026-09-20 12:30:22 +08:00
reformatting and styling
This commit is contained in:
@@ -18,7 +18,8 @@ def test_pwgan_generator():
|
||||
dropout=0.0,
|
||||
bias=True,
|
||||
use_weight_norm=True,
|
||||
upsample_factors=[4, 4, 4, 4])
|
||||
upsample_factors=[4, 4, 4, 4],
|
||||
)
|
||||
dummy_c = torch.rand((2, 80, 5))
|
||||
output = model(dummy_c)
|
||||
assert np.all(output.shape == (2, 1, 5 * 256)), output.shape
|
||||
|
||||
Reference in New Issue
Block a user