fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-05 09:10:25 +08:00
Code Activity

optimize(infer.synthesizer): all modules inherit from one

Browse Source
This commit is contained in:
源文雨
2024-06-10 21:34:35 +09:00
parent b67050b2f7
commit e33ef19200
5 changed files with 127 additions and 231 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
infer/modules/train/train.py
View File

@@ -415,6 +415,7 @@ def train_and_evaluate(
for batch_idx, info in data_iterator:
# Data
## Unpack
pitch = pitchf = None
if hps.if_f0 == 1:
(
phone,
@@ -444,22 +445,13 @@ def train_and_evaluate(
# Calculate
with autocast(enabled=hps.train.fp16_run):
if hps.if_f0 == 1:
(
y_hat,
ids_slice,
x_mask,
z_mask,
(z, z_p, m_p, logs_p, m_q, logs_q),
) = net_g(phone, phone_lengths, pitch, pitchf, spec, spec_lengths, sid)
else:
(
y_hat,
ids_slice,
x_mask,
z_mask,
(z, z_p, m_p, logs_p, m_q, logs_q),
) = net_g(phone, phone_lengths, spec, spec_lengths, sid)
(
y_hat,
ids_slice,
x_mask,
z_mask,
(z, z_p, m_p, logs_p, m_q, logs_q),
) = net_g(phone, phone_lengths, spec, spec_lengths, sid, pitch, pitchf)
mel = spec_to_mel_torch(
spec,
hps.data.filter_length,