Merge pull request #632 from SciML/ChrisRackauckas-patch-3

Use the parameterless type for doing adapting

Author: ChrisRackauckas

.buildkite/pipeline.yml

@@ -8,7 +8,7 @@ steps:
     agents:
       queue: "juliagpu"
       cuda: "*"
-    timeout_in_minutes: 120
+    timeout_in_minutes: 240
     # Don't run Buildkite if the commit message includes the text [skip tests]
     if: build.message !~ /\[skip tests\]/
 

src/training_strategies.jl

@@ -29,8 +29,8 @@ function merge_strategy_with_loss_function(pinnrep::PINNRepresentation,
 
     # the points in the domain and on the boundary
     pde_train_sets, bcs_train_sets = train_sets
-    pde_train_sets = adapt.(typeof(flat_init_params), pde_train_sets)
-    bcs_train_sets = adapt.(typeof(flat_init_params), bcs_train_sets)
+    pde_train_sets = adapt.(parameterless_type(ComponentArrays.getdata(flat_init_params)), pde_train_sets)
+    bcs_train_sets = adapt.(parameterless_type(ComponentArrays.getdata(flat_init_params)), bcs_train_sets)
     pde_loss_functions = [get_loss_function(_loss, _set, eltypeθ, strategy)
                           for (_loss, _set) in zip(datafree_pde_loss_function,
                                                    pde_train_sets)]
@@ -152,7 +152,7 @@ end
                                                     minibatch)
     lb, ub = bound
     set = QuasiMonteCarlo.generate_design_matrices(points, lb, ub, sampling_alg, minibatch)
-    set = map(s -> adapt(eltypeθ, s), set)
+    set = map(s -> adapt(parameterless_type(eltypeθ), s), set)
     return set
 end