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

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-21 08:28:34 +08:00
committed by GitHub
parent 345c6d8f54
commit 68d4b8d68b
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
train_ms.py
View File

@@ -92,14 +92,19 @@ def run(rank, n_gpus, hps):
else: else:
print("Using normal MAS for VITS1") print("Using normal MAS for VITS1")
use_noise_scaled_mas = False use_noise_scaled_mas = False
if "use_spk_conditioned_encoder" in hps.model.keys() and hps.model.use_spk_conditioned_encoder == True:
if hps.data.n_speakers == 0:
raise ValueError("n_speakers must be > 0 when using spk conditioned encoder to train multi-speaker model")
use_spk_conditioned_encoder = True
else:
print("Using normal encoder for VITS1")
use_spk_conditioned_encoder = False
net_g = SynthesizerTrn( net_g = SynthesizerTrn(
len(symbols), len(symbols),
hps.data.filter_length // 2 + 1, hps.data.filter_length // 2 + 1,
hps.train.segment_size // hps.data.hop_length, hps.train.segment_size // hps.data.hop_length,
n_speakers=hps.data.n_speakers, n_speakers=hps.data.n_speakers,
use_spk_conditioned_encoder=use_spk_conditioned_encoder,
use_noise_scaled_mas=use_noise_scaled_mas,
mas_noise_scale_initial = mas_noise_scale_initial, mas_noise_scale_initial = mas_noise_scale_initial,
noise_scale_delta = noise_scale_delta, noise_scale_delta = noise_scale_delta,
**hps.model).cuda(rank) **hps.model).cuda(rank)