From ca4b00f34049798706e128293c6fb54678148930 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Mon, 21 Aug 2023 08:24:23 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_ms.py b/train_ms.py index 7dc739e..64365f3 100644 --- a/train_ms.py +++ b/train_ms.py @@ -176,8 +176,8 @@ def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loade net_d.train() for batch_idx, (x, x_lengths, spec, spec_lengths, y, y_lengths, speakers, tone, language, bert) in tqdm(enumerate(train_loader)): if net_g.use_noise_scaled_mas: - current_mas_noise_scale = net_g.mas_noise_scale_initial - net_g.noise_scale_delta * global_step - net_g.current_mas_noise_scale = max(current_mas_noise_scale, 0.0) + current_mas_noise_scale = net_g.module.mas_noise_scale_initial - net_g.module.noise_scale_delta * global_step + net_g.module.current_mas_noise_scale = max(current_mas_noise_scale, 0.0) x, x_lengths = x.cuda(rank, non_blocking=True), x_lengths.cuda(rank, non_blocking=True) spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(rank, non_blocking=True) y, y_lengths = y.cuda(rank, non_blocking=True), y_lengths.cuda(rank, non_blocking=True)