Skip to content

Commit ab77299

Browse files
Add PDE Poisson to doc
1 parent 854b5f0 commit ab77299

File tree

2 files changed

+3
-9
lines changed

2 files changed

+3
-9
lines changed

docs/zh/api/equation.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,6 @@
99
- Laplace
1010
- NavierStokes
1111
- NormalDotVec
12+
- Poisson
1213
show_root_heading: false
1314
heading_level: 3

examples/darcy/darcy2d_v2.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -126,16 +126,9 @@ def poisson_ref_compute_func(_in):
126126

127127
# set training hyper-parameters
128128
epochs = 10000 if not args.epochs else args.epochs
129-
lr_scheduler = ppsci.optimizer.lr_scheduler.MultiStepDecay(
130-
epochs,
131-
ITERS_PER_EPOCH,
132-
0.001,
133-
(4000, 6000, 8000),
134-
0.1,
135-
)()
136129

137130
# set optimizer
138-
optimizer = ppsci.optimizer.Adam(lr_scheduler)((model,))
131+
optimizer = ppsci.optimizer.Adam(1e-3)((model,))
139132

140133
# set validator
141134
NPOINTS_EVAL = NPOINT_PDE
@@ -183,7 +176,7 @@ def poisson_ref_compute_func(_in):
183176
constraint,
184177
output_dir,
185178
optimizer,
186-
lr_scheduler,
179+
None,
187180
epochs,
188181
ITERS_PER_EPOCH,
189182
eval_during_train=True,

0 commit comments

Comments
 (0)