Make style

This commit is contained in:
Eren Gölge
2021-09-30 16:21:18 +00:00
parent 7edbe04fe0
commit 0b1986384f
+2 -3
View File
@@ -6,7 +6,6 @@ import torch
from coqpit import Coqpit
from torch import nn
# pylint: skip-file
@@ -80,7 +79,7 @@ class BaseModel(nn.Module, ABC):
...
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.
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
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()`"""
pass