From 63d6f52bef746c96b613c38cadd66b4916dcd64e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Wed, 6 Sep 2023 10:20:42 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_ms.py b/train_ms.py index 2e673b1..bc72253 100644 --- 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]: