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()