Skip to content

Commit 0d17d1a

Browse files
add norm clip for g training
1 parent 286369a commit 0d17d1a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train_ms.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -597,8 +597,8 @@ def train_and_evaluate(
597597
optim_g.zero_grad()
598598
scaler.scale(loss_gen_all).backward()
599599
scaler.unscale_(optim_g)
600-
if getattr(hps.train, "bf16_run", False):
601-
torch.nn.utils.clip_grad_norm_(parameters=net_g.parameters(), max_norm=500)
600+
#if getattr(hps.train, "bf16_run", False):
601+
torch.nn.utils.clip_grad_norm_(parameters=net_g.parameters(), max_norm=500)
602602
grad_norm_g = commons.clip_grad_value_(net_g.parameters(), None)
603603
scaler.step(optim_g)
604604
scaler.update()

0 commit comments

Comments
 (0)