Update PINNs_NavierStokes_example.ipynb

Same number of iterations for each example
cemac · Mar 7, 2024 · 65f5d0d · 65f5d0d
1 parent ba108c3
commit 65f5d0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PINNs_NavierStokes_example.ipynb b/PINNs_NavierStokes_example.ipynb
@@ -1081,7 +1081,7 @@
    "outputs": [],
    "source": [
     "# Training\n",
-    "train(sess, 20000, x_tf, y_tf, t_tf, u_tf, v_tf, x, y, t, u_train, v_train, loss, train_op_Adam, optimizer_Adam,\"modelckpts/NSn/\")"
+    "train(sess, Train_iterations, x_tf, y_tf, t_tf, u_tf, v_tf, x, y, t, u_train, v_train, loss, train_op_Adam, optimizer_Adam,\"modelckpts/NSn/\")"
    ]
   },
   {