tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix: tqdm output to sys.stdout

Browse Source
This commit is contained in:
litagin02
2023-12-30 00:06:26 +09:00
parent 98d9cda196
commit 82ef037b91
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
data_utils.py
View File

@@ -1,15 +1,18 @@
import os import os
import random import random
import sys
import numpy as np
import torch import torch
import torch.utils.data import torch.utils.data
from tqdm import tqdm from tqdm import tqdm
import numpy as np
from tools.log import logger
import commons import commons
from mel_processing import spectrogram_torch, mel_spectrogram_torch
from utils import load_wav_to_torch, load_filepaths_and_text
from text import cleaned_text_to_sequence
from config import config from config import config
from mel_processing import mel_spectrogram_torch, spectrogram_torch
from text import cleaned_text_to_sequence
from tools.log import logger
from utils import load_filepaths_and_text, load_wav_to_torch
"""Multi speaker version""" """Multi speaker version"""
@@ -61,7 +64,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
skipped = 0 skipped = 0
logger.info("Init dataset...") logger.info("Init dataset...")
for _id, spk, language, text, phones, tone, word2ph in tqdm( for _id, spk, language, text, phones, tone, word2ph in tqdm(
self.audiopaths_sid_text self.audiopaths_sid_text, file=sys.stdout
): ):
audiopath = f"{_id}" audiopath = f"{_id}"
if self.min_text_len <= len(phones) and len(phones) <= self.max_text_len: if self.min_text_len <= len(phones) and len(phones) <= self.max_text_len: