From c864828afa95833709149d6744cddbd1e3b50d3f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 7 Nov 2023 12:55:48 +0800 Subject: [PATCH] add emo --- train_ms.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/train_ms.py b/train_ms.py index 341dcca..67bd1dd 100644 --- a/train_ms.py +++ b/train_ms.py @@ -340,6 +340,7 @@ def train_and_evaluate( bert, ja_bert, en_bert, + emo, ) in tqdm(enumerate(train_loader)): if net_g.module.use_noise_scaled_mas: current_mas_noise_scale = ( @@ -362,6 +363,7 @@ def train_and_evaluate( bert = bert.cuda(rank, non_blocking=True) ja_bert = ja_bert.cuda(rank, non_blocking=True) en_bert = en_bert.cuda(rank, non_blocking=True) + emo = emo.cuda(rank, non_blocking=True) with autocast(enabled=hps.train.fp16_run): ( @@ -384,6 +386,7 @@ def train_and_evaluate( bert, ja_bert, en_bert, + emo, ) mel = spec_to_mel_torch( spec, @@ -579,6 +582,7 @@ def evaluate(hps, generator, eval_loader, writer_eval): bert, ja_bert, en_bert, + emo, ) in enumerate(eval_loader): x, x_lengths = x.cuda(), x_lengths.cuda() spec, spec_lengths = spec.cuda(), spec_lengths.cuda()