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

add emo

Browse Source
This commit is contained in:
Stardust·减
2023-11-07 12:55:48 +08:00
committed by GitHub
parent 59805037c1
commit c864828afa
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_ms.py
View File

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