linter fixes

This commit is contained in:
Eren Gölge 2021-04-12 13:14:11 +02:00
parent b11d1cb845
commit b735076bb4
4 changed files with 5 additions and 25 deletions

View File

@ -76,7 +76,7 @@ class Tacotron2(TacotronAbstract):
gst_style_tokens=10, gst_style_tokens=10,
gst_use_speaker_embedding=False, gst_use_speaker_embedding=False,
): ):
super(Tacotron2, self).__init__( super().__init__(
num_chars, num_chars,
num_speakers, num_speakers,
r, r,

View File

@ -4,7 +4,7 @@ from torch import nn
# pylint: disable=dangerous-default-value # pylint: disable=dangerous-default-value
class ResStack(nn.Module): class ResStack(nn.Module):
def __init__(self, kernel, channel, padding, dilations=[1, 3, 5]): def __init__(self, kernel, channel, padding, dilations=[1, 3, 5]):
super(ResStack, self).__init__() super().__init__()
resstack = [] resstack = []
for dilation in dilations: for dilation in dilations:
resstack += [ resstack += [

View File

@ -82,7 +82,7 @@ class MultiPeriodDiscriminator(torch.nn.Module):
""" """
def __init__(self): def __init__(self):
super(MultiPeriodDiscriminator, self).__init__() super().__init__()
self.discriminators = nn.ModuleList( self.discriminators = nn.ModuleList(
[ [
DiscriminatorP(2), DiscriminatorP(2),
@ -124,7 +124,7 @@ class DiscriminatorS(torch.nn.Module):
""" """
def __init__(self, use_spectral_norm=False): def __init__(self, use_spectral_norm=False):
super(DiscriminatorS, self).__init__() super().__init__()
norm_f = nn.utils.spectral_norm if use_spectral_norm else nn.utils.weight_norm norm_f = nn.utils.spectral_norm if use_spectral_norm else nn.utils.weight_norm
self.convs = nn.ModuleList( self.convs = nn.ModuleList(
[ [
@ -165,7 +165,7 @@ class MultiScaleDiscriminator(torch.nn.Module):
""" """
def __init__(self): def __init__(self):
super(MultiScaleDiscriminator, self).__init__() super().__init__()
self.discriminators = nn.ModuleList( self.discriminators = nn.ModuleList(
[ [
DiscriminatorS(use_spectral_norm=True), DiscriminatorS(use_spectral_norm=True),

View File

@ -1,20 +0,0 @@
set -e
TF_CPP_MIN_LOG_LEVEL=3
# # tests
nosetests tests -x &&\
# # runtime tests
./tests/test_demo_server.sh && \
./tests/test_resample.sh && \
./tests/test_tacotron_train.sh && \
./tests/test_glow-tts_train.sh && \
./tests/test_vocoder_gan_train.sh && \
./tests/test_vocoder_wavernn_train.sh && \
./tests/test_vocoder_wavegrad_train.sh && \
./tests/test_speedy_speech_train.sh && \
./tests/test_aligntts_train.sh && \
./tests/test_compute_statistics.sh && \
# linter check
cardboardlinter --refspec main