nshtrainer 1.0.0b19__py3-none-any.whl → 1.0.0b21__py3-none-any.whl

nshtrainer/trainer/_config.py

@@ -602,6 +602,10 @@ class TrainerConfig(C.Config):
     :meth:`~lightning.pytorch.core.LightningModule.log_dict`.
     """
 
+    experimental_barebones_profiler_enabled: bool = False
+    """If enabled, will enable the profiler in barebones mode.
+    The implementation of this is very hacky and may not work as expected."""
+
     precision: (
         Literal[
             "64-true",
@@ -822,6 +826,10 @@ class TrainerConfig(C.Config):
         yield from self.callbacks
 
     def _nshtrainer_all_logger_configs(self) -> Iterable[BaseLoggerConfig | None]:
+        # Disable all loggers if barebones mode is enabled
+        if self.barebones:
+            return
+
         yield from self.enabled_loggers()
         yield self.actsave_logger
 

nshtrainer/trainer/trainer.py

@@ -278,6 +278,12 @@ class Trainer(LightningTrainer):
         _update_kwargs(**hparams.lightning_kwargs)
         _update_kwargs(**kwargs_ctor)
 
+        # Handle barebones mode
+        if kwargs.get("barebones"):
+            # Remove the logger if it's an empty list
+            if (logger := kwargs.get("logger")) is not None and not logger:
+                kwargs["logger"] = None
+
         return kwargs
 
     if TYPE_CHECKING:

{nshtrainer-1.0.0b19.dist-info → nshtrainer-1.0.0b21.dist-info}/METADATA

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

{nshtrainer-1.0.0b19.dist-info → nshtrainer-1.0.0b21.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=2AIr8w_ysRtn0yo49rwdduyBJ9bIAVpQdRpJoMg9Cd0,33806
+nshtrainer/trainer/_config.py,sha256=xd80gDFSj5QvRIZ_2t15PN397akpXX8VUazeivS8kV0,34112
 nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
 nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
-nshtrainer/trainer/trainer.py,sha256=V5aRA6hBSxYi-Hbp-lg6b5mRCw_bc_0QzkJ7LG0c49M,19531
+nshtrainer/trainer/trainer.py,sha256=WydBYDuRpGFGkl180Wss6ADIu-fo_kQeDi-0LI64OyM,19770
 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.0b19.dist-info/METADATA,sha256=5WTNG0hJXpVLs3iZrpsIjgbccJeuIMG7i7t4MyAeIt4,937
-nshtrainer-1.0.0b19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-1.0.0b19.dist-info/RECORD,,
+nshtrainer-1.0.0b21.dist-info/METADATA,sha256=PwOjPxm1i_uXzfl6QBdjHcNnbwtbbXwgoIFxkBg7PFU,937
+nshtrainer-1.0.0b21.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b21.dist-info/RECORD,,