Make style

This commit is contained in:
Eren Gölge 2021-09-30 16:21:18 +00:00
parent 7edbe04fe0
commit 0b1986384f
1 changed files with 2 additions and 3 deletions

View File

@ -6,7 +6,6 @@ import torch
from coqpit import Coqpit from coqpit import Coqpit
from torch import nn from torch import nn
# pylint: skip-file # pylint: skip-file
@ -80,7 +79,7 @@ class BaseModel(nn.Module, ABC):
... ...
return outputs_dict, loss_dict return outputs_dict, loss_dict
def train_log(self, batch: Dict, outputs: Dict, logger: "Logger", assets:Dict, steps:int) -> None: def train_log(self, batch: Dict, outputs: Dict, logger: "Logger", assets: Dict, steps: int) -> None:
"""Create visualizations and waveform examples for training. """Create visualizations and waveform examples for training.
For example, here you can plot spectrograms and generate sample sample waveforms from these spectrograms to For example, here you can plot spectrograms and generate sample sample waveforms from these spectrograms to
@ -113,7 +112,7 @@ class BaseModel(nn.Module, ABC):
... ...
return outputs_dict, loss_dict return outputs_dict, loss_dict
def eval_log(self, batch: Dict, outputs: Dict, logger: "Logger", assets:Dict, steps:int) -> None: def eval_log(self, batch: Dict, outputs: Dict, logger: "Logger", assets: Dict, steps: int) -> None:
"""The same as `train_log()`""" """The same as `train_log()`"""
pass pass