nshtrainer 1.0.0b22__py3-none-any.whl → 1.0.0b24__py3-none-any.whl

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/trainer/trainer.py

@@ -319,6 +319,16 @@ class Trainer(LightningTrainer):
         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", True
+            )
+
         super().__init__(**kwargs)
 
         # Set up the profiler again
@@ -327,6 +337,12 @@ class Trainer(LightningTrainer):
 
             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())
 

{nshtrainer-1.0.0b22.dist-info → nshtrainer-1.0.0b24.dist-info}/METADATA

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

{nshtrainer-1.0.0b22.dist-info → nshtrainer-1.0.0b24.dist-info}/RECORD

@@ -119,10 +119,10 @@ nshtrainer/profiler/advanced.py,sha256=XrM3FX0ThCv5UwUrrH0l4Ow4LGAtpiBww2N8QAU5N
 nshtrainer/profiler/pytorch.py,sha256=8K37XvPnCApUpIK8tA2zNMFIaIiTLSoxKQoiyCPBm1Q,2757
 nshtrainer/profiler/simple.py,sha256=PimjqcU-JuS-8C0ZGHAdwCxgNLij4x0FH6WXsjBQzZs,1005
 nshtrainer/trainer/__init__.py,sha256=MmoydVS6aYeav7zgDAUHxAQrV_PMQsbnZTCuPnLH9Wk,128
-nshtrainer/trainer/_config.py,sha256=xd80gDFSj5QvRIZ_2t15PN397akpXX8VUazeivS8kV0,34112
+nshtrainer/trainer/_config.py,sha256=5tl4BJxrldAUj5bt9c89xH-lDail3lVn_y3jUPq_kE4,34320
 nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
 nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
-nshtrainer/trainer/trainer.py,sha256=SgAjALn8LsRWmvBPvHJOP0IKDZOKKl3lwfrC1q4J2EA,20134
+nshtrainer/trainer/trainer.py,sha256=Tr0ANJxgNVbuzqwbqwSKcIl1geXEPC9AlS0_YPwOdWw,20750
 nshtrainer/util/_environment_info.py,sha256=MT8mBe6ZolRfKiwU-les1P-lPNPqXpHQcfADrh_A3uY,24629
 nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
 nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
@@ -135,6 +135,6 @@ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
 nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
 nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
 nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
-nshtrainer-1.0.0b22.dist-info/METADATA,sha256=AkEEGt5cpSMgefZzzZWfTEu6DztnaodTTg9aOtiWQc8,937
-nshtrainer-1.0.0b22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-1.0.0b22.dist-info/RECORD,,
+nshtrainer-1.0.0b24.dist-info/METADATA,sha256=kJJwb8hMUOHfif2qiEB70hCSllpUOt48s-n8SRFkRJY,937
+nshtrainer-1.0.0b24.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b24.dist-info/RECORD,,