Update train_ms.py

2023-09-06 10:20:42 +08:00
parent 9e30c85a15
commit 63d6f52bef
1 changed files with 1 additions and 0 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -337,6 +337,7 @@ def evaluate(hps, generator, eval_loader, writer_eval):
            y, y_lengths = y.cuda(), y_lengths.cuda()
            speakers = speakers.cuda()
            bert = bert.cuda()
+            ja_bert = ja_bert.cuda()
            tone = tone.cuda()
            language = language.cuda()
            for use_sdp in [True, False]: