Yaxin
/
live-digital-avatar-gptsovits
25
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

s2_train_v3tqdm进度条修复

Browse Source 
s2_train_v3tqdm进度条修复
main
RVC-Boss 6 months ago 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

2
GPT_SoVITS/s2_train_v3.py
Unescape Escape 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

Write Preview
Loading…
Cancel
Save