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, bert,
ja_bert, ja_bert,
en_bert, en_bert,
emo,
) in tqdm(enumerate(train_loader)): ) in tqdm(enumerate(train_loader)):
if net_g.module.use_noise_scaled_mas: if net_g.module.use_noise_scaled_mas:
current_mas_noise_scale = ( current_mas_noise_scale = (
@@ -362,6 +363,7 @@ def train_and_evaluate(
bert = bert.cuda(rank, non_blocking=True) bert = bert.cuda(rank, non_blocking=True)
ja_bert = ja_bert.cuda(rank, non_blocking=True) ja_bert = ja_bert.cuda(rank, non_blocking=True)
en_bert = en_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): with autocast(enabled=hps.train.fp16_run):
( (
@@ -384,6 +386,7 @@ def train_and_evaluate(
bert, bert,
ja_bert, ja_bert,
en_bert, en_bert,
emo,
) )
mel = spec_to_mel_torch( mel = spec_to_mel_torch(
spec, spec,
@@ -579,6 +582,7 @@ def evaluate(hps, generator, eval_loader, writer_eval):
bert, bert,
ja_bert, ja_bert,
en_bert, en_bert,
emo,
) in enumerate(eval_loader): ) in enumerate(eval_loader):
x, x_lengths = x.cuda(), x_lengths.cuda() x, x_lengths = x.cuda(), x_lengths.cuda()
spec, spec_lengths = spec.cuda(), spec_lengths.cuda() spec, spec_lengths = spec.cuda(), spec_lengths.cuda()