wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-11 12:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

pylint fixes

Browse Source
This commit is contained in:
erogol
2020-10-27 12:35:10 +01:00
parent 3aeef5e83c
commit a6f564c8c8
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/layers/losses.py
+4 -3
View File
@@ -7,6 +7,8 @@ from torch.nn import functional
from TTS.tts.utils.generic_utils import sequence_mask from TTS.tts.utils.generic_utils import sequence_mask
# pylint: disable=abstract-method Method
# relates https://github.com/pytorch/pytorch/issues/42305
class L1LossMasked(nn.Module): class L1LossMasked(nn.Module):
def __init__(self, seq_len_norm): def __init__(self, seq_len_norm):
super().__init__() super().__init__()
@@ -145,9 +147,8 @@ class DifferentailSpectralLoss(nn.Module):
target_diff = target[:, 1:] - target[:, :-1] target_diff = target[:, 1:] - target[:, :-1]
if len(signature(self.loss_func).parameters) > 2: if len(signature(self.loss_func).parameters) > 2:
return self.loss_func(x_diff, target_diff, length-1) return self.loss_func(x_diff, target_diff, length-1)
else: # if loss masking is not enabled
# if loss masking is not enabled return self.loss_func(x_diff, target_diff)
return self.loss_func(x_diff, target_diff)
class GuidedAttentionLoss(torch.nn.Module): class GuidedAttentionLoss(torch.nn.Module):
TTS/vocoder/models/wavernn.py
+2 -1
View File
@@ -16,7 +16,8 @@ from TTS.vocoder.utils.distribution import (
def stream(string, variables): def stream(string, variables):
sys.stdout.write(f"\r{string}" % variables) sys.stdout.write(f"\r{string}" % variables)
# pylint: disable=abstract-method
# relates https://github.com/pytorch/pytorch/issues/42305
class ResBlock(nn.Module): class ResBlock(nn.Module):
def __init__(self, dims): def __init__(self, dims):
super().__init__() super().__init__()