mirror of https://github.com/coqui-ai/TTS.git
linter fix
This commit is contained in:
parent
53f54898bc
commit
4d3e1e9d9a
|
@ -1,6 +1,7 @@
|
|||
from torch import nn
|
||||
|
||||
|
||||
# pylint: disable=dangerous-default-value
|
||||
class ResStack(nn.Module):
|
||||
def __init__(self, kernel, channel, padding, dilations=[1, 3, 5]):
|
||||
super(ResStack, self).__init__()
|
||||
|
@ -40,7 +41,7 @@ class ResStack(nn.Module):
|
|||
|
||||
|
||||
class MRF(nn.Module):
|
||||
def __init__(self, kernels, channel, dilations=[1, 3, 5]):
|
||||
def __init__(self, kernels, channel, dilations=[1, 3, 5]): # # pylint: disable=dangerous-default-value
|
||||
super().__init__()
|
||||
self.resblock1 = ResStack(kernels[0], channel, 0, dilations)
|
||||
self.resblock2 = ResStack(kernels[1], channel, 6, dilations)
|
||||
|
|
|
@ -192,7 +192,7 @@ class HifiganGenerator(torch.nn.Module):
|
|||
self.resblocks = nn.ModuleList()
|
||||
for i in range(len(self.ups)):
|
||||
ch = upsample_initial_channel // (2**(i + 1))
|
||||
for j, (k, d) in enumerate(
|
||||
for _, (k, d) in enumerate(
|
||||
zip(resblock_kernel_sizes, resblock_dilation_sizes)):
|
||||
self.resblocks.append(resblock(ch, k, d))
|
||||
# post convolution layer
|
||||
|
|
|
@ -74,7 +74,7 @@ def gan_dataset_case(batch_size, seq_len, hop_len, conv_pad, return_pairs, retur
|
|||
check_item(feat2, wav2)
|
||||
count_iter += 1
|
||||
else:
|
||||
for item1 in loader:
|
||||
for item1 in loader:
|
||||
feat1, wav1 = item1
|
||||
check_item(feat1, wav1)
|
||||
count_iter += 1
|
||||
|
|
Loading…
Reference in New Issue