We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e9e4f6c commit 07e9f72Copy full SHA for 07e9f72
examples/chip_heat/chip_heat.py
@@ -388,8 +388,9 @@ def train(cfg: DictConfig):
388
top_sup_constraint.name: top_sup_constraint,
389
}
390
391
+ clip = paddle.nn.ClipGradByGlobalNorm(clip_norm=1.0)
392
# set optimizer
- optimizer = ppsci.optimizer.Adam(cfg.TRAIN.learning_rate)(model)
393
+ optimizer = ppsci.optimizer.Adam(cfg.TRAIN.learning_rate, grad_clip=clip)(model)
394
395
# set validator
396
top_down_label = {"chip": np.zeros([cfg.NL, 1], dtype="float32")}
0 commit comments