nshtrainer 1.0.0b21__tar.gz → 1.0.0b23__tar.gz

{nshtrainer-1.0.0b21 → nshtrainer-1.0.0b23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 1.0.0b21
+Version: 1.0.0b23
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-1.0.0b21 → nshtrainer-1.0.0b23}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "1.0.0-beta21"
+version = "1.0.0-beta23"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-1.0.0b21 → nshtrainer-1.0.0b23}/src/nshtrainer/trainer/_config.py

@@ -606,6 +606,10 @@ class TrainerConfig(C.Config):
     """If enabled, will enable the profiler in barebones mode.
     The implementation of this is very hacky and may not work as expected."""
 
+    experimental_barebones_progress_bar_enabled: bool = False
+    """If enabled, will enable the progress bar in barebones mode.
+    The implementation of this is very hacky and may not work as expected."""
+
     precision: (
         Literal[
             "64-true",

{nshtrainer-1.0.0b21 → nshtrainer-1.0.0b23}/src/nshtrainer/trainer/trainer.py

@@ -316,8 +316,33 @@ class Trainer(LightningTrainer):
         self._hparams = hparams
         self.debug = self.hparams.debug
 
+        experimental_profiler = None
+        if hparams.experimental_barebones_profiler_enabled:
+            experimental_profiler = kwargs.pop("profiler", None)
+            log.warning(
+                "Barebones profiler is enabled. This is an experimental feature and may not work as expected."
+            )
+
+        experimental_barebones_progress_bar = None
+        if hparams.experimental_barebones_progress_bar_enabled:
+            experimental_barebones_progress_bar = kwargs.pop(
+                "enable_progress_bar", None
+            )
+
         super().__init__(**kwargs)
 
+        # Set up the profiler again
+        if experimental_profiler is not None:
+            from lightning.pytorch.trainer import setup
+
+            setup._init_profiler(self, experimental_profiler)
+
+        # Set up the progress bar again
+        if experimental_barebones_progress_bar is not None:
+            self._callback_connector._configure_progress_bar(
+                experimental_barebones_progress_bar
+            )
+
         # Add our own start time callback to measure the start time.
         self.callbacks.append(RuntimeTrackerCallback())