fix the wrong invoke in SchrodingerTrainer

tum-pbs · Nov 13, 2024 · f505b9a · f505b9a
1 parent 5f4323a
commit f505b9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/PINN/lib_pinns/schrodinger/trainer.py b/experiments/PINN/lib_pinns/schrodinger/trainer.py
@@ -87,7 +87,7 @@ def _loss_funcs(self):
         else:
             return [self.get_internal_loss,self.get_boundary_loss,self.get_initial_loss]
 
-class BaseLineTrainer(SchrodingerTrainerBasis,BaseLineTrainerBasis):
+class StandardTrainer(SchrodingerTrainerBasis,StandardTrainerBasis):
     pass
 
 def training(epochs,trainer:SchrodingerTrainerBasis,