Skip to content

Commit 07e9f72

Browse files
chip_heat optimized (#1128)
* The ClipGradByGlobalNorm is added to the chip_heat.py to optimized the accuracy * pre-commit chip_heat.py
1 parent e9e4f6c commit 07e9f72

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/chip_heat/chip_heat.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -388,8 +388,9 @@ def train(cfg: DictConfig):
388388
top_sup_constraint.name: top_sup_constraint,
389389
}
390390

391+
clip = paddle.nn.ClipGradByGlobalNorm(clip_norm=1.0)
391392
# set optimizer
392-
optimizer = ppsci.optimizer.Adam(cfg.TRAIN.learning_rate)(model)
393+
optimizer = ppsci.optimizer.Adam(cfg.TRAIN.learning_rate, grad_clip=clip)(model)
393394

394395
# set validator
395396
top_down_label = {"chip": np.zeros([cfg.NL, 1], dtype="float32")}

0 commit comments

Comments
 (0)