mirror of https://github.com/coqui-ai/TTS.git
linter fixes #2
This commit is contained in:
parent
4d3e1e9d9a
commit
cd69da4868
|
@ -11,8 +11,7 @@ class MelganDiscriminator(nn.Module):
|
||||||
base_channels=16,
|
base_channels=16,
|
||||||
max_channels=1024,
|
max_channels=1024,
|
||||||
downsample_factors=(4, 4, 4, 4),
|
downsample_factors=(4, 4, 4, 4),
|
||||||
groups_denominator=4,
|
groups_denominator=4):
|
||||||
max_groups=256):
|
|
||||||
super(MelganDiscriminator, self).__init__()
|
super(MelganDiscriminator, self).__init__()
|
||||||
self.layers = nn.ModuleList()
|
self.layers = nn.ModuleList()
|
||||||
|
|
||||||
|
|
|
@ -89,4 +89,3 @@ def test_melgan_feature_loss():
|
||||||
loss_func = MelganFeatureLoss()
|
loss_func = MelganFeatureLoss()
|
||||||
loss = loss_func(feats_fake, feats_real)
|
loss = loss_func(feats_fake, feats_real)
|
||||||
assert loss.item() == 0
|
assert loss.item() == 0
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue