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(rvc.utils): more type defs & rename

Browse Source
This commit is contained in:
源文雨
2024-06-07 19:33:45 +09:00
parent c10c527264
commit 49488dcae9
6 changed files with 41 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -481,7 +481,7 @@ def train_and_evaluate(
optim_d.zero_grad()
scaler.scale(loss_disc).backward()
scaler.unscale_(optim_d)
grad_norm_d = utils.clip_grad_value_(net_d.parameters(), None)
grad_norm_d = utils.total_grad_norm(net_d.parameters())
scaler.step(optim_d)
with autocast(enabled=hps.train.fp16_run):
@@ -496,7 +496,7 @@ def train_and_evaluate(
optim_g.zero_grad()
scaler.scale(loss_gen_all).backward()
scaler.unscale_(optim_g)
grad_norm_g = utils.clip_grad_value_(net_g.parameters(), None)
grad_norm_g = utils.total_grad_norm(net_g.parameters())
scaler.step(optim_g)
scaler.update()