Skip to content

Commit 46a012d

Browse files
authored
docs: change update step to thousands in PINN 2D PDE (#1153)
For 50,000 training steps, an update every 1000 step is enough detail
1 parent 367680b commit 46a012d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/PINN2DPDE/main.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ function train_model(xyt, target_data, xyt_bc, target_bc; seed::Int=0,
180180

181181
isnan(loss) && throw(ArgumentError("NaN Loss Detected"))
182182

183-
if iter % 500 == 1 || iter == maxiters
183+
if iter % 1000 == 1 || iter == maxiters
184184
@printf "Iteration: [%5d / %5d] \t Loss: %.9f (%.9f) \t Physics Loss: %.9f \
185185
(%.9f) \t Data Loss: %.9f (%.9f) \t BC \
186186
Loss: %.9f (%.9f)\n" iter maxiters loss mean_loss stats.physics_loss mean_physics_loss stats.data_loss mean_data_loss stats.bc_loss mean_bc_loss

0 commit comments

Comments
 (0)