formatting

This commit is contained in:
Eren Gölge 2021-04-08 11:13:55 +02:00
parent aee24b0704
commit 15f362d5b1
2 changed files with 1 additions and 1 deletions

View File

@ -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.

View File

@ -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