Save/Load of Optimizer State (#33)

* Enable save/load of optimizer; Save/load manager like model * Add training settings as mutable settings to Manager * Save/load optimizer when new one is available; svae total num steps * Save/load optimizer when new one is available; svae total num steps * Add removed training writer back in * Revert additional kwargs for now --------- Co-authored-by: mhuen <mhuen@users.noreply.github.com>
icecube · Dec 19, 2024 · e93b142 · e93b142
1 parent 59226a4
commit e93b142
Show file tree

Hide file tree

Showing 5 changed files with 280 additions and 236 deletions.
diff --git a/egenerator/ic3/reconstruction.py b/egenerator/ic3/reconstruction.py
@@ -399,10 +399,10 @@ def Configure(self):
 
         for model in self.manager.models:
             num_vars, num_total_vars = model.num_variables
-            msg = "\nNumber of Model Variables:\n"
-            msg += "\tFree: {}\n"
-            msg += "\tTotal: {}"
-            log_info(msg.format(num_vars, num_total_vars))
+            msg = f"\nNumber of Model Variables for {model.name}:\n"
+            msg += f"\tFree: {num_vars}\n"
+            msg += f"\tTotal: {num_total_vars}"
+            log_info(msg)
 
         # ------------------------------
         # Gather Reconstruction Settings