add scheduler step for multisolvers (#526)

2025-03-27 15:55:15 +01:00
parent b958c0f5db
commit f48da47ed4
3 changed files with 17 additions and 10 deletions
--- a/pina/solver/physics_informed_solver/self_adaptive_pinn.py
+++ b/pina/solver/physics_informed_solver/self_adaptive_pinn.py
@@ -188,12 +188,14 @@ class SelfAdaptivePINN(PINNInterface, MultiSolverInterface):
        loss = super().training_step(batch)
        self.manual_backward(-loss)
        self.optimizer_weights.instance.step()
+        self.scheduler_weights.instance.step()

        # Model optimization
        self.optimizer_model.instance.zero_grad()
        loss = super().training_step(batch)
        self.manual_backward(loss)
        self.optimizer_model.instance.step()
+        self.scheduler_model.instance.step()

        return loss