Skip to content

Commit 0a94c48

Browse files
Update train_ms.py
1 parent 0dcb83f commit 0a94c48

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_ms.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -598,7 +598,7 @@ def train_and_evaluate(
598598
scaler.scale(loss_gen_all).backward()
599599
scaler.unscale_(optim_g)
600600
#if getattr(hps.train, "bf16_run", False):
601-
torch.nn.utils.clip_grad_norm_(parameters=net_g.parameters(), max_norm=500)
601+
torch.nn.utils.clip_grad_norm_(parameters=net_g.parameters(), max_norm=200)
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)