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
|
from torch import nn
|
||||||
|
|
||||||
|
|
||||||
|
# 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(ResStack, self).__init__()
|
||||||
|
@ -40,7 +41,7 @@ class ResStack(nn.Module):
|
||||||
|
|
||||||
|
|
||||||
class MRF(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__()
|
super().__init__()
|
||||||
self.resblock1 = ResStack(kernels[0], channel, 0, dilations)
|
self.resblock1 = ResStack(kernels[0], channel, 0, dilations)
|
||||||
self.resblock2 = ResStack(kernels[1], channel, 6, dilations)
|
self.resblock2 = ResStack(kernels[1], channel, 6, dilations)
|
||||||
|
|
|
@ -192,7 +192,7 @@ class HifiganGenerator(torch.nn.Module):
|
||||||
self.resblocks = nn.ModuleList()
|
self.resblocks = nn.ModuleList()
|
||||||
for i in range(len(self.ups)):
|
for i in range(len(self.ups)):
|
||||||
ch = upsample_initial_channel // (2**(i + 1))
|
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)):
|
zip(resblock_kernel_sizes, resblock_dilation_sizes)):
|
||||||
self.resblocks.append(resblock(ch, k, d))
|
self.resblocks.append(resblock(ch, k, d))
|
||||||
# post convolution layer
|
# 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)
|
check_item(feat2, wav2)
|
||||||
count_iter += 1
|
count_iter += 1
|
||||||
else:
|
else:
|
||||||
for item1 in loader:
|
for item1 in loader:
|
||||||
feat1, wav1 = item1
|
feat1, wav1 = item1
|
||||||
check_item(feat1, wav1)
|
check_item(feat1, wav1)
|
||||||
count_iter += 1
|
count_iter += 1
|
||||||
|
|
Loading…
Reference in New Issue