blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

s2_train_v3tqdm进度条修复

Browse Source 
s2_train_v3tqdm进度条修复
This commit is contained in:
RVC-Boss 2025-02-12 14:54:09 +08:00 committed by GitHub
parent 8e9e3c07d5
commit 86acb7a89d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/s2_train_v3.py
View File

@ -281,7 +281,7 @@ def train_and_evaluate(
# text, # text,
# text_lengths, # text_lengths,
# ) in enumerate(tqdm(train_loader)): # ) in enumerate(tqdm(train_loader)):
for batch_idx, (ssl, spec, mel, ssl_lengths, spec_lengths, text, text_lengths, mel_lengths) in tqdm(enumerate(train_loader)): for batch_idx, (ssl, spec, mel, ssl_lengths, spec_lengths, text, text_lengths, mel_lengths) in enumerate(tqdm(train_loader)):
if torch.cuda.is_available(): if torch.cuda.is_available():
spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda( spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(
rank, non_blocking=True rank, non_blocking=True