linter fix

This commit is contained in:
Eren Gölge 2021-04-08 14:57:46 +02:00
parent 53f54898bc
commit 4d3e1e9d9a
3 changed files with 4 additions and 3 deletions

View File

@ -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)

View File

@ -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

View File

@ -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