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

Make style and lint

Browse Source
This commit is contained in:
Eren Gölge
2021-12-01 10:42:52 +00:00
parent 40cb8ac966
commit ce45d9e1af
3 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/bin/collect_env_info.py
+2
View File
@@ -8,8 +8,10 @@ import torch
sys.path += [os.path.abspath(".."), os.path.abspath(".")]
import json
import TTS
def system_info():
return {
"OS": platform.system(),
TTS/utils/callbacks.py
+14 -10
View File
@@ -1,8 +1,6 @@
class TrainerCallback:
def __init__(self):
super().__init__()
def on_init_start(self, trainer) -> None:
@staticmethod
def on_init_start(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_init_start"):
trainer.model.module.on_init_start(trainer)
@@ -16,7 +14,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_init_start"):
trainer.optimizer.on_init_start(trainer)
def on_init_end(self, trainer) -> None:
@staticmethod
def on_init_end(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_init_end"):
trainer.model.module.on_init_end(trainer)
@@ -30,7 +29,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_init_end"):
trainer.optimizer.on_init_end(trainer)
def on_epoch_start(self, trainer) -> None:
@staticmethod
def on_epoch_start(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_epoch_start"):
trainer.model.module.on_epoch_start(trainer)
@@ -44,7 +44,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_epoch_start"):
trainer.optimizer.on_epoch_start(trainer)
def on_epoch_end(self, trainer) -> None:
@staticmethod
def on_epoch_end(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_epoch_end"):
trainer.model.module.on_epoch_end(trainer)
@@ -58,7 +59,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_epoch_end"):
trainer.optimizer.on_epoch_end(trainer)
def on_train_step_start(self, trainer) -> None:
@staticmethod
def on_train_step_start(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_train_step_start"):
trainer.model.module.on_train_step_start(trainer)
@@ -72,7 +74,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_train_step_start"):
trainer.optimizer.on_train_step_start(trainer)
def on_train_step_end(self, trainer) -> None:
@staticmethod
def on_train_step_end(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_train_step_end"):
trainer.model.module.on_train_step_end(trainer)
@@ -86,7 +89,8 @@ class TrainerCallback:
if hasattr(trainer.optimizer, "on_train_step_end"):
trainer.optimizer.on_train_step_end(trainer)
def on_keyboard_interrupt(self, trainer) -> None:
@staticmethod
def on_keyboard_interrupt(trainer) -> None:
if hasattr(trainer.model, "module"):
if hasattr(trainer.model.module, "on_keyboard_interrupt"):
trainer.model.module.on_keyboard_interrupt(trainer)
TTS/vocoder/models/gan.py
+1 -1
View File
@@ -310,7 +310,7 @@ class GAN(BaseVocoder):
data_items: List,
verbose: bool,
num_gpus: int,
rank:int=0
rank: int = 0, # pylint: disable=unused-argument
):
"""Initiate and return the GAN dataloader.