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

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-09-06 10:20:42 +08:00
committed by GitHub
parent 9e30c85a15
commit 63d6f52bef
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train_ms.py
View File

@@ -337,6 +337,7 @@ def evaluate(hps, generator, eval_loader, writer_eval):
y, y_lengths = y.cuda(), y_lengths.cuda() y, y_lengths = y.cuda(), y_lengths.cuda()
speakers = speakers.cuda() speakers = speakers.cuda()
bert = bert.cuda() bert = bert.cuda()
ja_bert = ja_bert.cuda()
tone = tone.cuda() tone = tone.cuda()
language = language.cuda() language = language.cuda()
for use_sdp in [True, False]: for use_sdp in [True, False]: