mirror of
https://github.com/wassname/TTS.git
synced 2026-09-10 11:50:20 +08:00
formatting
This commit is contained in:
@@ -33,6 +33,7 @@ class TorchSTFT(nn.Module): # pylint: disable=abstract-method
|
||||
self.mel_basis = None
|
||||
if use_mel:
|
||||
self._build_mel_basis()
|
||||
|
||||
@torch.no_grad()
|
||||
def __call__(self, x):
|
||||
"""Compute spectrogram frames by torch based stft.
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
# adopted from https://github.com/jik876/hifi-gan/blob/master/models.py
|
||||
|
||||
import torch
|
||||
from torch import nn
|
||||
from torch.nn import functional as F
|
||||
|
||||
Reference in New Issue
Block a user