I’m using NeuralPDE.jl to solve a system of nonlinear PDEs, following the example in the docs.
I wonder if there is a way to add the iteration number in the callback function, so I will know the progress.
The callback function I’m using from the docs is:
callback = function (p, l)
println("loss: ", l)
println("pde_losses: ", map(l_ -> l_(p), pde_inner_loss_functions))
println("bcs_losses: ", map(l_ -> l_(p), bcs_inner_loss_functions))
return false
end
Which shows the different loss functions, but not the iteration number.
Thank you!