Skip to content

Commit 3aefe25

Browse files
committed
Amend example to match paper results
1 parent bd24d40 commit 3aefe25

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/burgers-new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,9 @@ def f_model(u_model, x, t):
3838

3939
model = CollocationSolverND()
4040
model.compile(layer_sizes, f_model, Domain, BCs)
41-
model.fit(newton_iter=301)
4241

43-
model.fit(tf_iter=301)
42+
# to reproduce results from Raissi and the SA-PINNs paper, train for 10k newton and 10k adam
43+
model.fit(newton_iter=10000, tf_iter=10000)
4444

4545

4646
#######################################################

0 commit comments

Comments
 (0)