eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clean paired_voice

Browse Source
This commit is contained in:
James Betker 2022-01-06 10:26:53 -07:00
parent 525addffab
commit 5e1d1da2e9
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/data/audio/paired_voice_audio_dataset.py
View File

@ -7,13 +7,9 @@ import torch
import torch.nn.functional as F import torch.nn.functional as F
import torch.utils.data import torch.utils.data
import torchaudio import torchaudio
from munch import munchify
from tokenizers import Tokenizer
from tqdm import tqdm from tqdm import tqdm
from transformers import GPT2TokenizerFast
from data.audio.unsupervised_audio_dataset import load_audio, load_similar_clips from data.audio.unsupervised_audio_dataset import load_audio, load_similar_clips
from data.util import find_files_of_type, is_audio_file
from models.tacotron2.taco_utils import load_filepaths_and_text from models.tacotron2.taco_utils import load_filepaths_and_text
from models.tacotron2.text import text_to_sequence, sequence_to_text from models.tacotron2.text import text_to_sequence, sequence_to_text
from utils.util import opt_get from utils.util import opt_get
@ -222,7 +218,7 @@ if __name__ == '__main__':
batch_sz = 8 batch_sz = 8
params = { params = {
'mode': 'paired_voice_audio', 'mode': 'paired_voice_audio',
'path': ['Y:\\bigasr_dataset\\hifi_tts_mp3\\test.txt'], 'path': ['Y:\\bigasr_dataset\\hifi_tts\\test.txt'],
'fetcher_mode': ['libritts'], 'fetcher_mode': ['libritts'],
'phase': 'train', 'phase': 'train',
'n_workers': 0, 'n_workers': 0,