diff --git a/generate_data_properties.py b/generate_data_properties.py index c8f78ae..92ffd0a 100644 --- a/generate_data_properties.py +++ b/generate_data_properties.py @@ -4,6 +4,7 @@ Parameters from hparam.py will be used """ + import argparse import os import sys diff --git a/hifigandenoiser.py b/hifigandenoiser.py index b0c364a..c249b55 100644 --- a/hifigandenoiser.py +++ b/hifigandenoiser.py @@ -1,4 +1,5 @@ """Waveglow style denoiser can be used to remove the artifacts from the HiFiGAN generated audio.""" + import torch from src.utilities.stft import STFT diff --git a/src/data_module.py b/src/data_module.py index 947cb03..80837b5 100644 --- a/src/data_module.py +++ b/src/data_module.py @@ -3,6 +3,7 @@ Contains PyTorch-Lightning's datamodule and dataloaders """ + import nltk import pytorch_lightning as pl from torch.utils.data import DataLoader diff --git a/src/hparams.py b/src/hparams.py index e570192..559ba2c 100644 --- a/src/hparams.py +++ b/src/hparams.py @@ -3,6 +3,7 @@ Hyper Parameters for the experiment """ + import os from argparse import Namespace diff --git a/src/model/HMMComponents/Decoder.py b/src/model/HMMComponents/Decoder.py index 9065485..1fdfaca 100644 --- a/src/model/HMMComponents/Decoder.py +++ b/src/model/HMMComponents/Decoder.py @@ -1,5 +1,6 @@ """Parameter Model This model takes state as an input and generates its parameters i.e mean, standard deviation and the probability of transition to the next state.""" + import torch import torch.nn as nn import torch.nn.functional as F diff --git a/src/model/HMMComponents/TransitionModel.py b/src/model/HMMComponents/TransitionModel.py index 0e8ede4..8ef3d62 100644 --- a/src/model/HMMComponents/TransitionModel.py +++ b/src/model/HMMComponents/TransitionModel.py @@ -3,6 +3,7 @@ This network is responsible for the models response of whether to switch to next state or keep emitting at the current state. """ + import torch import torch.nn as nn diff --git a/src/model/layers.py b/src/model/layers.py index ef1b786..90519f6 100644 --- a/src/model/layers.py +++ b/src/model/layers.py @@ -2,6 +2,7 @@ Layer modules used in the model design """ + import torch import torch.nn as nn from librosa.filters import mel as librosa_mel_fn diff --git a/src/training_module.py b/src/training_module.py index a2f2bca..beee149 100644 --- a/src/training_module.py +++ b/src/training_module.py @@ -4,6 +4,7 @@ This file contains PyTorch Lightning's main module where code of the main model is implemented """ + import os from argparse import Namespace diff --git a/src/utilities/data.py b/src/utilities/data.py index b0df91c..0352248 100644 --- a/src/utilities/data.py +++ b/src/utilities/data.py @@ -3,6 +3,7 @@ Utilities for processing of Data """ + import random import numpy as np diff --git a/src/utilities/functions.py b/src/utilities/functions.py index 8658d69..1db68c5 100644 --- a/src/utilities/functions.py +++ b/src/utilities/functions.py @@ -2,6 +2,7 @@ File for custom utility functions to improve numerical precision """ + import torch diff --git a/src/utilities/plotting.py b/src/utilities/plotting.py index 0062f3f..e0c20bb 100644 --- a/src/utilities/plotting.py +++ b/src/utilities/plotting.py @@ -3,6 +3,7 @@ File contains utilities for plotting """ + from typing import Any import matplotlib diff --git a/src/utilities/text/__init__.py b/src/utilities/text/__init__.py index a1b61f5..750cc45 100644 --- a/src/utilities/text/__init__.py +++ b/src/utilities/text/__init__.py @@ -1,4 +1,5 @@ """from https://github.com/keithito/tacotron.""" + import re from src.utilities.text import cleaners diff --git a/train.py b/train.py index 278f836..1f1659b 100644 --- a/train.py +++ b/train.py @@ -3,6 +3,7 @@ PyTorch-Lightning Trainer file, main file to run your training """ + import argparse import os