We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 286369a commit 0d17d1aCopy full SHA for 0d17d1a
train_ms.py
@@ -597,8 +597,8 @@ def train_and_evaluate(
597
optim_g.zero_grad()
598
scaler.scale(loss_gen_all).backward()
599
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)
+ #if getattr(hps.train, "bf16_run", False):
+ torch.nn.utils.clip_grad_norm_(parameters=net_g.parameters(), max_norm=500)
602
grad_norm_g = commons.clip_grad_value_(net_g.parameters(), None)
603
scaler.step(optim_g)
604
scaler.update()
0 commit comments