remove PINO code

YichengDWu · Aug 21, 2023 · 8d254c4 · 8d254c4
1 parent bb8af16
commit 8d254c4
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 176 deletions.
diff --git a/docs/src/tutorials/burgers.jl b/docs/src/tutorials/burgers.jl
diff --git a/src/pde/discretize.jl b/src/pde/discretize.jl
@@ -61,36 +61,6 @@ function build_loss_function(pde_system::PDESystem, pinn::PINN,
     return pde_and_bcs_loss_function
 end
 
-#=
-function build_loss_function(pde_system::ParametricPDESystem, pinn::PINN,
-                             strategy::AbstractTrainingAlg, coord_branch_net;
-                             derivative=finitediff)
-    (; eqs, bcs, ivs, dvs, pvs) = pde_system
-    (; phi, init_params) = pinn
-
-    depvars, indvars, dict_indvars, dict_depvars, dict_depvar_input = get_vars(ivs, dvs)
-    _, _, _, dict_pmdepvars, dict_pmdepvar_input = get_vars(ivs, pvs)
-
-    multioutput = false
-
-    pinnrep = (; eqs, bcs, depvars, indvars, dict_indvars, dict_depvars, dict_depvar_input,
-               dict_pmdepvars, dict_pmdepvar_input, multioutput, pvs, init_params, pinn,
-               derivative, strategy, fdtype, coord_branch_net)
-
-    datafree_pde_loss_functions = Tuple(build_loss_function(pinnrep, first(eq), i)
-                                        for (i, eq) in enumerate(eqs))
-
-    datafree_bc_loss_functions = Tuple(build_loss_function(pinnrep, first(bc),
-                                                           i +
-                                                           length(datafree_pde_loss_functions))
-                                       for (i, bc) in enumerate(bcs))
-
-    pde_and_bcs_loss_function = scalarize(strategy, phi, datafree_pde_loss_functions,
-                                          datafree_bc_loss_functions)
-    return pde_and_bcs_loss_function
-end
-=#
-
 """
      discretize(pde_system::PDESystem, pinn::PINN, sampler::PINNSampler,
                     strategy::AbstractTrainingAlg; derivative=finitediff,
@@ -122,37 +92,6 @@ function discretize(pde_system, pinn::PINN, sampler::PINNSampler,
     return Optimization.OptimizationProblem(f, init_params, datasets)
 end
 
-# ParametricPDESystem no long supported
-#=function discretize(pde_system::ParametricPDESystem, pinn::PINN, sampler::PINNSampler,
-                    strategy::AbstractTrainingAlg, functionsampler::FunctionSampler,
-                    coord_branch_net::AbstractArray;
-                    additional_loss=Sophon.null_additional_loss, derivative=finitediff,
-                    fdtype=Float64,
-                    adtype=Optimization.AutoZygote())
-    datasets = sample(pde_system, sampler)
-    init_params = Lux.fmap(Base.Fix1(broadcast, fdtype), pinn.init_params)
-    init_params = _ComponentArray(init_params)
-
-    datasets = map(Base.Fix1(broadcast, fdtype), datasets)
-    datasets = init_params isa AbstractGPUComponentVector ?
-               map(Base.Fix1(adapt, CuArray), datasets) : datasets
-
-    pfs = sample(functionsampler)
-    coord_branch_net = coord_branch_net isa Union{AbstractVector, StepRangeLen} ?
-                      [coord_branch_net] : coord_branch_net
-    pde_and_bcs_loss_function = build_loss_function(pde_system, pinn, strategy,
-                                                    coord_branch_net, derivative,
-                                                    fdtype)
-    function full_loss_function(θ, p)
-        return pde_and_bcs_loss_function(θ, p) + additional_loss(pinn.phi, θ)
-    end
-    f = OptimizationFunction(full_loss_function, adtype)
-
-    p = PINOParameterHandler(datasets, pfs)
-    return Optimization.OptimizationProblem(f, init_params, p)
-end
-=#
-
 function symbolic_discretize(pde_system, pinn::PINN, sampler::PINNSampler,
                              strategy::AbstractTrainingAlg;
                              additional_loss=Sophon.null_additional_loss, derivative=finitediff,

diff --git a/src/pde/pinn_types.jl b/src/pde/pinn_types.jl
@@ -230,18 +230,3 @@ function Base.show(io::IO, ::MIME"text/plain", sys::ParametricPDESystem)
     println(io, "Parametric Variables: ", sys.pvs)
     return nothing
 end
-
-mutable struct PINOParameterHandler
-    coords::Any
-    fs::Any
-end
-
-get_local_ps(p::PINOParameterHandler) = p.coords
-get_global_ps(p::PINOParameterHandler) = p.fs
-Base.getindex(p::PINOParameterHandler, i) = getindex(p.coords, i)
-
-@inline get_local_ps(p::Vector{<:AbstractMatrix}) = p
-@inline get_global_ps(::Vector{<:AbstractMatrix}) = nothing
-
-ChainRulesCore.@non_differentiable get_local_ps(::Any...)
-ChainRulesCore.@non_differentiable get_global_ps(::Any...)