@@ -35,14 +35,14 @@ import Lux: tanh, identity
3535 end
3636
3737 res = Optimization. solve (prob, Adam (0.01 ); callback = callback, maxiters = 500 )
38- prob = remake (prob, u0 = res. minimizer )
38+ prob = remake (prob, u0 = res. u )
3939 res = Optimization. solve (prob, Adam (0.001 ); callback = callback, maxiters = 200 )
4040 phi = discretization. phi
4141
4242 xs, ys = [infimum (d. domain): 0.01 : supremum (d. domain) for d in domains]
4343 analytic_sol_func (x, y) = (sin (pi * x) * sin (pi * y)) / (2pi ^ 2 )
4444
45- u_predict = reshape ([first (phi ([x, y], res. minimizer )) for x in xs for y in ys],
45+ u_predict = reshape ([first (phi ([x, y], res. u )) for x in xs for y in ys],
4646 (length (xs), length (ys)))
4747 u_real = reshape ([analytic_sol_func (x, y) for x in xs for y in ys],
4848 (length (xs), length (ys)))
8787 end
8888
8989 res = Optimization. solve (prob, Adam (0.01 ); callback = callback, maxiters = 300 )
90- prob = remake (prob, u0 = res. minimizer )
90+ prob = remake (prob, u0 = res. u )
9191 res = Optimization. solve (prob, Adam (0.001 ); callback = callback, maxiters = 300 )
9292 phi = discretization. phi
9393
103103 xs = collect (infimum (domains2[2 ]. domain): 1.0 : supremum (domains2[2 ]. domain))
104104
105105 u_real= [analytic_sol_func (t,x) for t in ts, x in xs]
106- u_predict= [first (phi ([t, x], res. minimizer )) for t in ts, x in xs]
106+ u_predict= [first (phi ([t, x], res. u )) for t in ts, x in xs]
107107 @test u_predict ≈ u_real rtol= 0.05
108108end
109109
154154 res = Optimization. solve (prob, Adam (0.01 ); callback = callback, maxiters = 300 );
155155 phi = discretization. phi;
156156
157- u_predict= [first (phi ([t, x], res. minimizer )) for t in ts, x in xs]
157+ u_predict= [first (phi ([t, x], res. u )) for t in ts, x in xs]
158158
159159 @test u_predict ≈ u_MOL rtol= 0.025
160160
0 commit comments