convert

SciML · Sep 23, 2024 · 5677ade · 5677ade
1 parent 5c34346
commit 5677ade
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/discretize.jl b/src/discretize.jl
@@ -219,19 +219,21 @@ function generate_training_sets(domains, dx, eqs, bcs, eltypeθ, dict_indvars::D
         span = map(b -> get(dict_var_span, b, b), bt)
         _set = adapt(eltypeθ,
             hcat(vec(map(points -> collect(points), Iterators.product(span...)))...))
+        x = convert.(eltypeθ, _set)
     end
 
     pde_vars = get_variables(eqs, dict_indvars, dict_depvars)
     pde_args = get_argument(eqs, dict_indvars, dict_depvars)
 
-    pde_train_set = adapt(eltypeθ,
-        hcat(vec(map(points -> collect(points),
-            Iterators.product(bc_data...)))...))
+    # pde_train_set = adapt(eltypeθ,
+    #     hcat(vec(map(points -> collect(points),
+    #         Iterators.product(bc_data...)))...))
 
     pde_train_sets = map(pde_args) do bt
         span = map(b -> get(dict_var_span_, b, b), bt)
         _set = adapt(eltypeθ,
             hcat(vec(map(points -> collect(points), Iterators.product(span...)))...))
+        x = convert.(eltypeθ, _set)
     end
     [pde_train_sets, bcs_train_sets]
 end

diff --git a/src/neural_adapter.jl b/src/neural_adapter.jl
@@ -7,6 +7,7 @@ function generate_training_sets(domains, dx, eqs, eltypeθ)
     spans = [infimum(d.domain):dx:supremum(d.domain) for (d, dx) in zip(domains, dxs)]
     train_set = adapt(eltypeθ,
         hcat(vec(map(points -> collect(points), Iterators.product(spans...)))...))
+    convert.(eltypeθ, train_set)
 end
 
 function get_loss_function_(loss, init_params, pde_system, strategy::GridTraining)