From 316e0ba345f8d15925e2553db0f2081ddaf42ff2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 09:57:08 +0800 Subject: [PATCH] add tqdm --- data_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/data_utils.py b/data_utils.py index 08906dc..0dfe191 100644 --- a/data_utils.py +++ b/data_utils.py @@ -4,6 +4,8 @@ import random import numpy as np import torch import torch.utils.data +from tqdm import tqdm + import commons from mel_processing import spectrogram_torch, mel_spectrogram_torch, spec_to_mel_torch from utils import load_wav_to_torch, load_filepaths_and_text @@ -55,7 +57,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset): audiopaths_sid_text_new = [] lengths = [] skipped = 0 - for _id, spk, language, text, phones, tone, word2ph in self.audiopaths_sid_text: + for _id, spk, language, text, phones, tone, word2ph in tqdm(self.audiopaths_sid_text): audiopath = f'{_id}' if self.min_text_len <= len(phones) and len(phones) <= self.max_text_len: phones = phones.split(" ")