rename operator_loss to inital_condition_loss

SciML · Jun 12, 2024 · bb0863b · bb0863b
1 parent cc1fada
commit bb0863b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pino_ode_solve.jl b/src/pino_ode_solve.jl
@@ -102,7 +102,7 @@ function physics_loss(
  sum(abs2, du .- f_) / norm
 end
 
-function operator_loss(
+function inital_condition_loss(
  phi::PINOPhi{C, T}, prob::ODEProblem, x, θ) where {C <: DeepONet, T}
  p, t = x
  f = prob.f
@@ -129,7 +129,7 @@ end
 function generate_loss(strategy::GridTraining, prob::ODEProblem, phi, bounds, tspan)
  x = get_trainset(strategy, bounds, tspan)
  function loss(θ, _)
- operator_loss(phi, prob, x, θ) + physics_loss(phi, prob, x, θ)
+ inital_condition_loss(phi, prob, x, θ) + physics_loss(phi, prob, x, θ)
  end
 end