File tree Expand file tree Collapse file tree 2 files changed +3
-9
lines changed Expand file tree Collapse file tree 2 files changed +3
-9
lines changed Original file line number Diff line number Diff line change 9
9
- Laplace
10
10
- NavierStokes
11
11
- NormalDotVec
12
+ - Poisson
12
13
show_root_heading: false
13
14
heading_level: 3
Original file line number Diff line number Diff line change @@ -126,16 +126,9 @@ def poisson_ref_compute_func(_in):
126
126
127
127
# set training hyper-parameters
128
128
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
- )()
136
129
137
130
# set optimizer
138
- optimizer = ppsci .optimizer .Adam (lr_scheduler )((model ,))
131
+ optimizer = ppsci .optimizer .Adam (1e-3 )((model ,))
139
132
140
133
# set validator
141
134
NPOINTS_EVAL = NPOINT_PDE
@@ -183,7 +176,7 @@ def poisson_ref_compute_func(_in):
183
176
constraint ,
184
177
output_dir ,
185
178
optimizer ,
186
- lr_scheduler ,
179
+ None ,
187
180
epochs ,
188
181
ITERS_PER_EPOCH ,
189
182
eval_during_train = True ,
You can’t perform that action at this time.
0 commit comments