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 random
import torch import torch
import torchaudio
from torch.utils.data.dataset import random_split from torch.utils.data.dataset import random_split
from torchaudio.datasets import LJSPEECH, LIBRITTS from torchaudio.datasets import LJSPEECH, LIBRITTS
from torchaudio.transforms import MuLawEncoding from torchaudio.transforms import MuLawEncoding
......
import argparse import argparse
import torch import torch
import torch.nn.functional as F
import torchaudio import torchaudio
from torchaudio.transforms import MelSpectrogram from torchaudio.transforms import MelSpectrogram
from torchaudio.models import wavernn from torchaudio.models import wavernn
......
import argparse import argparse
import logging import logging
import os import os
import signal
from collections import defaultdict from collections import defaultdict
from datetime import datetime from datetime import datetime
from time import time from time import time
...@@ -9,7 +8,6 @@ from typing import List ...@@ -9,7 +8,6 @@ from typing import List
import torch import torch
import torchaudio import torchaudio
from torch import nn as nn
from torch.optim import Adam from torch.optim import Adam
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from torchaudio.datasets.utils import bg_iterator from torchaudio.datasets.utils import bg_iterator
......
...@@ -34,6 +34,7 @@ from processing import ( ...@@ -34,6 +34,7 @@ from processing import (
bits_to_normalized_waveform, bits_to_normalized_waveform,
) )
class WaveRNNInferenceWrapper(torch.nn.Module): class WaveRNNInferenceWrapper(torch.nn.Module):
def __init__(self, wavernn: WaveRNN): 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