Unverified Commit 2e58f18a authored by yangarbiter's avatar yangarbiter Committed by GitHub
Browse files

Refactor coding style for wavernn example (#1663)

parent 077a5f4a
import os
import random
import torch
import torchaudio
from torch.utils.data.dataset import random_split
from torchaudio.datasets import LJSPEECH, LIBRITTS
from torchaudio.transforms import MuLawEncoding
......
import argparse
import torch
import torch.nn.functional as F
import torchaudio
from torchaudio.transforms import MelSpectrogram
from torchaudio.models import wavernn
......
import argparse
import logging
import os
import signal
from collections import defaultdict
from datetime import datetime
from time import time
......@@ -9,7 +8,6 @@ from typing import List
import torch
import torchaudio
from torch import nn as nn
from torch.optim import Adam
from torch.utils.data import DataLoader
from torchaudio.datasets.utils import bg_iterator
......
......@@ -34,6 +34,7 @@ from processing import (
bits_to_normalized_waveform,
)
class WaveRNNInferenceWrapper(torch.nn.Module):
def __init__(self, wavernn: WaveRNN):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment