* Fixing mean tracked loss

* Adding a PINA progress bar
2024-08-06 12:31:01 +02:00
parent 0fa4e1e58a
commit cce9876751
6 changed files with 194 additions and 36 deletions
--- a/pina/callbacks/init.py
+++ b/pina/callbacks/init.py
@@ -1,5 +1,10 @@
-__all__ = ["SwitchOptimizer", "R3Refinement", "MetricTracker"]
+__all__ = [
+    "SwitchOptimizer",
+    "R3Refinement",
+    "MetricTracker",
+    "PINAProgressBar"
+    ]

 from .optimizer_callbacks import SwitchOptimizer
 from .adaptive_refinment_callbacks import R3Refinement
-from .processing_callbacks import MetricTracker
+from .processing_callbacks import MetricTracker, PINAProgressBar