linter fixes #2

TTS/bin/train_vocoder_gan.py

@@ -479,7 +479,7 @@ def main(args):  # pylint: disable=redefined-outer-name
     optimizer_gen = getattr(torch.optim, c.optimizer)
     optimizer_gen = optimizer_gen(lr=c.lr_gen, **c.optimizer_params)
     optimizer_disc = getattr(torch.optim, c.optimizer)
-    optimizer_disc= optimizer_disc(lr=c.lr_gen, **c.optimizer_params)
+    optimizer_disc = optimizer_disc(lr=c.lr_gen, **c.optimizer_params)
 
     # schedulers
     scheduler_gen = None

TTS/vocoder/datasets/gan_dataset.py

@@ -120,7 +120,7 @@ class GANDataset(Dataset):
             else:
                 audio = self.ap.load_wav(wavpath)
                 mel = np.load(feat_path)
-                audio, mel= self._pad_short_samples(audio, mel)
+                audio, mel = self._pad_short_samples(audio, mel)
 
         # correct the audio length wrt padding applied in stft
         audio = np.pad(audio, (0, self.hop_len), mode="edge")

TTS/vocoder/layers/hifigan.py

@@ -56,4 +56,4 @@ class MRF(nn.Module):
     def remove_weight_norm(self):
         self.resblock1.remove_weight_norm()
         self.resblock2.remove_weight_norm()
-        self.resblock3.remove_weight_norm()
+        self.resblock3.remove_weight_norm()

TTS/vocoder/models/melgan_discriminator.py

@@ -11,8 +11,7 @@ class MelganDiscriminator(nn.Module):
                  base_channels=16,
                  max_channels=1024,
                  downsample_factors=(4, 4, 4, 4),
-                 groups_denominator=4,
-                 max_groups=256):
+                 groups_denominator=4):
         super(MelganDiscriminator, self).__init__()
         self.layers = nn.ModuleList()
 

tests/test_vocoder_losses.py

@@ -89,4 +89,3 @@ def test_melgan_feature_loss():
     loss_func = MelganFeatureLoss()
     loss = loss_func(feats_fake, feats_real)
     assert loss.item() == 0
-